Update app.py

app.py

@@ -1,20 +1,18 @@
-from diffusers import StableVideoDiffusionPipeline
 import torch
-from PIL import Image
+from diffusers import StableVideoDiffusionPipeline
+
+# CPU पर लोड करने के लिए
+pipeline = StableVideoDiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid")  
+pipeline.to("cpu", torch.float32)  # ✅ GPU की जगह CPU का इस्तेमाल
 
-# Model Load
-pipe = StableVideoDiffusionPipeline.from_pretrained(
-    "stabilityai/stable-video-diffusion-img2vid",
-    torch_dtype=torch.float16
-).to("cuda")
+# इनफरेंस फंक्शन (वीडियो जनरेट करना)
+def generate_video(prompt):
+    video = pipeline(prompt).videos  # ✅ अब यह CPU पर रन करेगा
+    return video
 
-def generate_video(image_path, prompt):
-    image = Image.open(image_path).convert("RGB")  # Load image properly
-    video = pipe(prompt=prompt, image=image, num_inference_steps=25).frames
-    video_path = "generated_video.mp4"
-    video[0].save(video_path)  # Save the first generated video
-    return video_path
+# टेस्टिंग
+prompt = "A cute cat playing with a ball"
+output_video = generate_video(prompt)
 
-# Example Usage
-video_file = generate_video("sample_image.jpg", "A beautiful mountain landscape with a sunset")
-print(f"Video generated and saved at: {video_file}")
+# आउटपुट सेव करें
+output_video[0].save("generated_video.mp4")  # ✅ CPU पर वीडियो सेव होगा