.

options/Video_model/Model.py

@@ -3,6 +3,8 @@ import torch,os,imageio
 from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import load_image, export_to_video
 from PIL import Image
+from glob import glob
+from pathlib import Path
 import numpy as np
 
 # Check if CUDA (GPU) is available, otherwise use CPU
@@ -37,13 +39,11 @@ def Video(image):
     # Set random seed for reproducibility
     generator = torch.manual_seed(42)
     
-    # Ensure the image is moved to the appropriate device (GPU or CPU)
-    # image = image.to(device)
-    
+    os.makedirs(output_folder, exist_ok=True)
+    base_count = len(glob(os.path.join(output_folder, "*.mp4")))
+    video_path = os.path.join(output_folder, f"{base_count:06d}.mp4")
     # Generate the video frames
     frames = pipeline(image, decode_chunk_size=8, generator=generator).frames[0]
-    path="generated.mp4"
     # Export the frames to a video file
-    save_video(frames, path, fps=7)
-    # vid=cv2.VideoCapture(path)
-    return path
+    export_to_video(frames, video_path, fps=fps_id)
+    return video_path