Update app.py

app.py

@@ -7,47 +7,34 @@ import gradio as gr
 
 import os
 HF_TOKEN = os.getenv("HF_TOKEN")
+ 
 
+# 1. Load pipeline
+pipe = StableVideoDiffusionPipeline.from_pretrained(
+    "stabilityai/stable-video-diffusion-img2vid",
+    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+    use_safetensors=True,
+    token=HF_TOKEN
+)
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = pipe.to(device)
 
-def generate_video(prompt_img_url):
-    pipe = StableVideoDiffusionPipeline.from_pretrained(
-        "stabilityai/stable-video-diffusion-img2vid",
-        torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-        use_safetensors=True,
-        token=HF_TOKEN
-    )
+# 2. Load input image
+img_url = "https://image.pollinations.ai/prompt/cat%20running"
+img = Image.open(requests.get(img_url, stream=True).raw).convert("RGB")
+img = img.resize((576, 320))
 
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-    pipe = pipe.to(device)
+# 3. Generate video frames
+frames = pipe(img, num_frames=5).frames[0]
 
-    # 2. Load input image
-    img = Image.open(requests.get(prompt_img_url, stream=True).raw).convert("RGB")
-    img = img.resize((576, 320))
+# 4. Save frames
+os.makedirs("frames", exist_ok=True)
+for i, f in enumerate(frames):
+    f.save(f"frames/frame_{i:03d}.png")
 
-    # 3. Generate video frames
-    frames = pipe(img, num_frames=5).frames[0]
+# 5. Build MP4 video
+os.system("ffmpeg -y -framerate 10 -i frames/frame_%03d.png output.mp4")
 
-    # 4. Save frames
-    os.makedirs("frames", exist_ok=True)
+print("Video saved as output.mp4")
 
-    for i, f in enumerate(frames):
-        f.save(f"frames/frame_{i:03d}.png")
-
-    # 5. Build MP4
-    os.system("ffmpeg -y -framerate 10 -i frames/frame_%03d.png output.mp4")
-
-    return "output.mp4"
-
-
-# Gradio UI
-with gr.Blocks() as demo:
-    gr.Markdown("# Stable Video Diffusion (IMG2VID)")
-
-    img_url = gr.Textbox(label="Image URL", value="https://image.pollinations.ai/prompt/cat%20running")
-    btn = gr.Button("Generate video")
-    output_video = gr.Video()
-
-    btn.click(generate_video, inputs=img_url, outputs=output_video)
-
-demo.launch()