Update app.py

app.py

@@ -12,7 +12,7 @@ import spaces
 import torch
 from PIL import Image
 from io import BytesIO
-from diffusers import AutoencoderKL, DiffusionPipeline, StableDiffusionImg2ImgPipeline, AutoPipelineForImage2Image
+from diffusers import AutoencoderKL, DiffusionPipeline, StableDiffusionImg2ImgPipeline, StableDiffusionXLPipeline
 
 DESCRIPTION = "# Run any LoRA or SD Model"
 if not torch.cuda.is_available():
@@ -61,7 +61,7 @@ def generate(
     if torch.cuda.is_available():
 
         if use_img2img:
-            pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
+            pipe = StableDiffusionXLPipeline.from_pretrained(model, torch_dtype=torch.float16)
             
             response = requests.get(url)
             init_image = Image.open(BytesIO(response.content)).convert("RGB")