Update app.py

app.py

@@ -12,8 +12,8 @@ def infer(source_img, prompt, negative_prompt, height, width, guide, steps, seed
     generator = torch.Generator(device).manual_seed(seed)     
     source_image = resize(height, width, source_img)
     source_image.save('source.png')
-    model_url = "https://huggingface.co/DmatryMakeev/anna-asti-2/resolve/main/ANNA_ASTI-2.ckpt"
-    pipe = DiffusionPipeline.from_pretrained(model_url, torch_dtype=torch.float16) if torch.cuda.is_available() else DiffusionPipeline.from_pretrained("DmatryMakeev/anna-asti-2", dtype=torch.float16)
+    model_url = "stabilityai/stable-diffusion-xl-base-1.0"
+    pipe = DiffusionPipeline.from_pretrained(model_url, torch_dtype=torch.float16) if torch.cuda.is_available() else DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
     pipe = pipe.to(device)
     image = pipe(prompt, negative_prompt=negative_prompt, image=source_image, strength=strength, guidance_scale=guide, num_inference_steps=steps).images[0]
     return image