Update app.py

app.py

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