Update app.py

app.py

@@ -25,11 +25,12 @@ def text_to_speech(text, output_path="output.mp3"):
 
 # 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"
+    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)["sample"][0]
+        image = pipe(prompt, num_inference_steps=50).images[0]
         image_path = f"image_{i}.png"
         image.save(image_path)
         images.append(image_path)
@@ -56,7 +57,7 @@ def create_video(images, audio_path, output_video_path="output.mp4"):
 def generate_video_from_prompt(prompt):
     script = generate_script(prompt)
     audio_path = text_to_speech(script)
-    images = generate_images_from_prompt(script)
+    images = generate_images_from_prompt(prompt)
     create_video(images, audio_path)
     return "output.mp4"