Update app.py

app.py

@@ -5,9 +5,9 @@ from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 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("cuda")
+
 def diffusion(text,num_inference_steps,guidance_scale):
     prompt = text
     image = pipe(prompt,guidance_scale=guidance_scale, num_inference_steps=num_inference_steps).images[0]