Update app.py

app.py

@@ -19,15 +19,16 @@ import streamlit as st
 
 def pic_mo(prom):
     model_id = "runwayml/stable-diffusion-v1-5"
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-    pipe = pipe.to("cpu")
+    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)  # Change to float32
+    # No need to specify .to("cuda") if running on CPU
 
     prompt = str(prom)
 
-    image = pipe(prompt).images[0]
-
+    image = pipe(prompt).images[0]  
+    
     return image
 
+
 def main():
     st.title("Text to Image with runwayml/stable-diffusion-v1-5")