Update app.py

app.py

@@ -6,8 +6,8 @@ from diffusers import StableDiffusionPipeline
 
 # "stabilityai/stable-diffusion-2-1-base"
 # "CompVis/stable-diffusion-v1-4"
-def create_model(loc = "stabilityai/stable-diffusion-2-1-base", mch = 'cuda'):
-    pipe = StableDiffusionPipeline.from_pretrained(loc, torch_dtype=torch.float16)
+def create_model(loc = "stabilityai/stable-diffusion-2-1-base", mch = 'cpu'):
+    pipe = StableDiffusionPipeline.from_pretrained(loc)
     pipe = pipe.to(mch)
     return pipe