tentative attempt at using ZeroGPU

app.py

@@ -3,6 +3,7 @@ import os
 import sys
 import tempfile
 import shutil
+import spaces
 from pathlib import Path
 import torch
 import logging
@@ -480,6 +481,27 @@ def initialize_inference_pipeline():
     logger.info("Inference pipeline initialized successfully")
     return inference_pipeline
 
+def get_duration(reference_image,
+                    audio_file, 
+                    text_prompt,
+                    seed,
+                    use_random_seed,
+                    num_steps,
+                    guidance_scale,
+                    audio_scale,
+                    overlap_frames,
+                    fps,
+                    silence_duration,
+                    resolution="720p"):
+    if num_steps > 15:
+        return 120
+    else:
+        return 100
+
+
+# this task might too long for ZeroGPU maybe,
+# but I need to try on a H200 to be sure
+@spaces.GPU(duration=get_duration)
 def generate_avatar_video(
     reference_image,
     audio_file,