Update app.py

app.py

@@ -43,7 +43,7 @@ def infer(prompt, negative, scale):
     model_id = base_dir 
     # Use the Euler scheduler here instead
     scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float32)
+    pipe = DiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, use_safetensors=True)
     # pipe = pipe.to("cuda")
     images = pipe(prompt).images
     return images