Update app.py

app.py

@@ -1,15 +1,16 @@
-import gradio as gr
 from diffusers import StableDiffusionPipeline
 import ffmpeg
 import os
+import gradio as gr
 
-# Load the Stable Diffusion model
-model = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-lite")
+# Load the Stable Diffusion 3.5 Medium model
+model = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3.5-medium")
+model.to("cuda")  # Use GPU for faster processing (if available)
 
 def generate_video(prompt):
     # Generate frames
     frames = []
-    for i in range(5):  # Generate 5 frames
+    for i in range(5):  # Adjust the number of frames for your video
         image = model(prompt).images[0]
         frame_path = f"frame_{i}.png"
         image.save(frame_path)
@@ -19,16 +20,16 @@ def generate_video(prompt):
     output_video = "output.mp4"
     (
         ffmpeg
-        .input("frame_%d.png", framerate=1)  # Adjust framerate for longer/shorter videos
+        .input("frame_%d.png", framerate=1)  # Adjust framerate
         .output(output_video)
         .run(overwrite_output=True)
     )
 
-    # Clean up temporary frame files
+    # Clean up frames
     for frame in frames:
         os.remove(frame)
 
-    return output_video  # Return the path to the video file
+    return output_video  # Path to the generated video
 
 # Gradio interface
 with gr.Blocks() as demo: