Update app.py

app.py

@@ -1,27 +1,25 @@
-import torch
-from PIL import Image
 import gradio as gr
 from diffusers import StableVideoDiffusionPipeline
+import torch
 
-# ✅ Model Load (Low Memory Mode for CPU)
-pipeline = StableVideoDiffusionPipeline.from_pretrained(
+# Model Load करो (CPU के लिए)
+pipe = StableVideoDiffusionPipeline.from_pretrained(
     "stabilityai/stable-video-diffusion-img2vid",
-    torch_dtype=torch.float32  # ✅ float16 से float32 में बदलें
-).to("cpu")  # ✅ CPU Mode पर चलाएं
+    torch_dtype=torch.float32
+)
 
+# Function जो Image से Video Generate करेगा
 def generate_video(image):
-    """AI Model से Image → Video Generate करे"""
-    video = pipeline(image, num_inference_steps=10).frames
-    video_path = "generated_video.mp4"
-    video[0].save(video_path)  # ✅ Video को Save करें
-    return video_path
+    video = pipe(image).frames
+    return video
 
-# ✅ Gradio UI (API को Web Interface में बदले)
-demo = gr.Interface(
+# Gradio Interface बनाओ
+iface = gr.Interface(
     fn=generate_video,
-    inputs=gr.Image(type="pil"),  # ✅ Image Input चाहिए
-    outputs=gr.Video()
+    inputs=gr.Image(label="Upload an image"),
+    outputs=gr.Video(label="Generated Video"),
+    title="AI Video Generator",
+    description="Upload an image to generate an AI video."
 )
 
-if __name__ == "__main__":
-    demo.launch()
+iface.launch()