Update app.py

app.py

@@ -4,11 +4,10 @@ import torch
 from diffusers import StableDiffusionPipeline
 
 st.title(" New App")
-
+pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16")
+pipe.to("cpu")
 area = st.text_area("Area",'')
 if len(area)>5:
-    pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float32)
-    
     image = pipe(area).images[0]
     im = Image.open(image)
     st.image(im)