Update app.py

app.py

@@ -233,7 +233,7 @@ def generate_video(prompt, seed, image=None):
         task_type = TaskType.I2V
         model_id = "Skywork/SkyReels-V1-Hunyuan-I2V"
         seed = 43
-        generator = torch.Generator(device="cuda").manual_seed(seed)
+        #generator = torch.Generator(device="cuda").manual_seed(seed)
 
         kwargs = {
             "prompt": prompt,
@@ -243,7 +243,7 @@ def generate_video(prompt, seed, image=None):
             "num_frames": 97,
             "num_inference_steps": 30,
             "seed": seed,
-            "generator": generator,
+            #"generator": generator,
             "guidance_scale": 6.0,
             "embedded_guidance_scale": 1.0,
             "negative_prompt": "Aerial view, low quality, bad hands",
@@ -267,7 +267,7 @@ def generate_video(prompt, seed, image=None):
     out_samples = []
     with torch.no_grad():
         output = _predictor.infer(**kwargs)
-        out_samples.extend(output.frames[0])
+        #out_samples.extend(output.frames[0])
     #output = (output.cpu().numpy() * 255).astype(np.uint8)
     #output = output.transpose(0, 2, 3, 4, 1)
 
@@ -275,7 +275,7 @@ def generate_video(prompt, seed, image=None):
     os.makedirs(save_dir, exist_ok=True)
     video_out_file = f"{save_dir}/{seed}.mp4"
     print(f"generate video, local path: {video_out_file}")
-    export_to_video(out_samples, video_out_file, fps=24)
+    export_to_video(output, video_out_file, fps=24)
     return video_out_file, kwargs
 
 def create_gradio_interface():