Update app.py

app.py

@@ -52,7 +52,7 @@ def generate(
     use_vae: bool = False,
     use_lora: bool = False,
     apply_refiner: bool = False,
-    model = 'runwayml/stable-diffusion-v1-5',
+    model = 'cagliostrolab/animagine-xl-3.0',
     vaecall = 'stabilityai/sd-vae-ft-mse',
     lora = 'amazonaws-la/juliette',
     lora_scale: float = 0.7,
@@ -60,7 +60,7 @@ def generate(
     if torch.cuda.is_available():
 
         if not use_vae:  
-            pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model, torch_dtype=torch.float16)
+            pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
             
         if use_vae:
             vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)