Update app.py

app.py

@@ -17,11 +17,8 @@ def modelo2(text):
     model_id = "stabilityai/stable-diffusion-2-1"
 
     # Use the DPMSolverMultistepScheduler (DPM-Solver++) scheduler here instead
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
     pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
-    pipe = pipe.to("cpu")
-    print(text)
- # Conversión a float32
     image = pipe(text).images[0]
 
     return image