Update app.py

app.py

@@ -16,8 +16,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 
-repo = "stabilityai/stable-diffusion-3-medium-diffusers"
-pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16, vae=vae).to(device)
+repo = "fluently/Fluently-XL-Final"
+pipe = StableDiffusionXLPipeline.from_pretrained(repo, torch_dtype=torch.float16, vae=vae).to(device)
 
 help_text = """
 To optimize image results:
@@ -43,7 +43,7 @@ def set_timesteps_patched(self, num_inference_steps: int, device = None):
     self.sigmas = self.sigmas.to("cpu") 
 
 # Image Editor
-edit_file = hf_hub_download(repo_id="stabilityai/cosxl", filename="cosxl_edit.safetensors")
+edit_file = hf_hub_download(repo, filename="cosxl_edit.safetensors")
 EDMEulerScheduler.set_timesteps = set_timesteps_patched
 pipe_edit = StableDiffusionXLInstructPix2PixPipeline.from_single_file(
     edit_file, num_in_channels=8, is_cosxl_edit=True, vae=vae, torch_dtype=torch.float16,