fn=generate_video,

app.py

@@ -45,8 +45,31 @@ def initialize_model(model_path):
     print('Model initialized: ' + model_path)
     return hunyuan_video_sampler
 
-@spaces.GPU(duration=120)
 def generate_video(
+    prompt,
+    resolution,
+    video_length,
+    seed,
+    num_inference_steps,
+    guidance_scale,
+    flow_shift,
+    embedded_guidance_scale
+):
+    print('generate_video (prompt: ' + prompt + ')')
+    return generate_video_gpu(
+        model,
+        prompt,
+        resolution,
+        video_length,
+        seed,
+        num_inference_steps,
+        guidance_scale,
+        flow_shift,
+        embedded_guidance_scale
+    )
+
+@spaces.GPU(duration=120)
+def generate_video_gpu(
     model,
     prompt,
     resolution,
@@ -57,6 +80,7 @@ def generate_video(
     flow_shift,
     embedded_guidance_scale
 ):
+    print('generate_video_gpu (prompt: ' + prompt + ')')
     if torch.cuda.device_count() == 0:
         gr.Warning('Set this space to GPU config to make it work.')
         return None
@@ -65,6 +89,7 @@ def generate_video(
     width, height = resolution.split("x")
     width, height = int(width), int(height)
     negative_prompt = "" # not applicable in the inference
+    print('Predicting video...')
 
     outputs = model.predict(
         prompt=prompt,
@@ -81,6 +106,7 @@ def generate_video(
         embedded_guidance_scale=embedded_guidance_scale
     )
     
+    print('Video predicted')
     samples = outputs['samples']
     sample = samples[0].unsqueeze(0)
     
@@ -92,6 +118,7 @@ def generate_video(
     save_videos_grid(sample, video_path, fps=24)
     logger.info(f'Sample saved to: {video_path}')
     
+    print('Return the video')
     return video_path
 
 def create_demo(model_path):
@@ -158,7 +185,7 @@ If you can't use _Hunyuan Video_, you can use _[CogVideoX](https://huggingface.c
                     """)
         
         generate_btn.click(
-            fn=lambda *inputs: generate_video(model, *inputs),
+            fn=generate_video,
             inputs=[
                 prompt,
                 resolution,