Create Video_generator.py

Video_generator.py

@@ -0,0 +1,14 @@
+from diffusers import StableVideoDiffusionPipeline
+import torch
+
+# Model Load
+pipe = StableVideoDiffusionPipeline.from_pretrained(
+    "stabilityai/stable-video-diffusion-img2vid",
+    torch_dtype=torch.float16
+).to("cuda")
+
+def generate_video(prompt):
+    video = pipe(prompt, num_inference_steps=25).frames
+    video_path = "generated_video.mp4"
+    video[0].save(video_path)  # Save the first generated video
+    return video_path