Update app.py

app.py

@@ -4,22 +4,19 @@ import torch
 import spaces
 import random
 from PIL import Image
-#import numpy as np
+import numpy as np
 
 from glob import glob
 from pathlib import Path
 from typing import Optional
 
-from diffsynth import ModelManager, SVDVideoPipeline, HunyuanDiTImagePipeline
-from diffsynth import ModelManager
-from diffusers.utils import load_image, export_to_video
+from diffsynth import save_video, ModelManager, SVDVideoPipeline, HunyuanDiTImagePipeline
 
 import uuid
 
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 # Constants
-#MAX_SEED = np.iinfo(np.int32).max
-MAX_SEED= 2147483647
+MAX_SEED = np.iinfo(np.int32).max
 CSS = """
 footer {
     visibility: hidden;
@@ -66,7 +63,7 @@ def generate(
     base_count = len(glob(os.path.join(output_folder, "*.mp4")))
     video_path = os.path.join(output_folder, f"{base_count:06d}.mp4")
 
-    frames = pipe(
+    video = pipe(
         input_image=image.resize((512, 512)), 
         num_frames=num_frames, 
         fps=fps_id, 
@@ -79,7 +76,8 @@ def generate(
         contrast_enhance_scale=1.2
     )
     model_manager.to("cpu")
-    export_to_video(frames, video_path, fps=fps_id)
+    
+    save_video(video, video_path, fps=fps_id)
     
     return video_path, seed