Update app.py

app.py

@@ -23,13 +23,13 @@ def text_to_speech(text, output_path="output.mp3"):
     tts.save(output_path)
     return output_path
 
-# Function to generate images from a text prompt using Stable Diffusion
+# Function to generate images from a text prompt using Stable Diffusion (CPU)
 def generate_images_from_prompt(prompt, num_images=5):
     model_id = "CompVis/stable-diffusion-v1-4"
-    pipe = StableDiffusionPipeline.from_pretrained(model_id).to("cuda")
+    pipe = StableDiffusionPipeline.from_pretrained(model_id)
     images = []
     for i in range(num_images):
-        image = pipe(prompt).images[0]
+        image = pipe(prompt, num_inference_steps=50)["sample"][0]
         image_path = f"image_{i}.png"
         image.save(image_path)
         images.append(image_path)