Update app.py

app.py

@@ -4,13 +4,10 @@ import torch
 
 MODEL_ID = "hakurei/waifu-diffusion-v1-4"
 
-def generate(prompt):
-    pipe = StableDiffusionPipeline.from_pretrained(
-        MODEL_ID,
-        torch_dtype=torch.float16
-    )
-    pipe = pipe.to("cpu")
+pipe = StableDiffusionPipeline.from_pretrained(MODEL_ID)
+pipe = pipe.to("cpu")
 
+def generate(prompt):
     image = pipe(prompt, num_inference_steps=15).images[0]
     return image