.

options/Video_model/Model.py

@@ -1,4 +1,4 @@
-import torch,os
+import torch,os,cv2
 from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import load_image, export_to_video
 from PIL import Image
@@ -38,4 +38,4 @@ def Video(image):
     # Export the frames to a video file
     export_to_video(frames, path, fps=7)
     
-    return path
+    return cv2.VideoCapture(path)

requirements.txt

@@ -8,4 +8,5 @@ torch==2.1.0
 torchvision==0.16.0
 transformers==4.42.3
 xformers==0.0.22.post7
-peft==0.9.0
+peft==0.9.0
+opencv-python