Update app.py

app.py

@@ -9,7 +9,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Load the pipeline
 pipeline = StableVideoDiffusionPipeline.from_pretrained(
-    "stabilityai/stable-video-diffusion-img2vid-xt", torch_dtype=torch.float16, variant="fp16"
+    "stabilityai/stable-video-diffusion-img2vid-xt-1-1", torch_dtype=torch.float16, variant="fp16"
 )
 pipeline.to(device)
 
@@ -27,7 +27,7 @@ def generate_video(image_path, seed):
 
     # Export the frames to a video file
     output_video_path = "generated.mp4"
-    export_to_video(frames, output_video_path, fps=7)
+    export_to_video(frames, output_video_path, fps=25)
 
     return output_video_path
 
@@ -43,5 +43,7 @@ iface = gr.Interface(
     description="Generate a video from an uploaded image using Stable Video Diffusion.",
 )
 
+example_image = "image.jpeg"
+example_video = "videos/generated.mp4"
 # Launch the interface
 iface.launch()