Update app.py

app.py

@@ -263,7 +263,23 @@ print("=" * 80)
 print("Pipeline fully loaded and ready!")
 print("=" * 80)
 
-@spaces.GPU(duration=80)
+def get_duration(
+    input_image,
+    prompt,
+    duration,
+    enhance_prompt,
+    seed,
+    randomize_seed,
+    height,
+    width,
+    progress)
+):
+    if duration <= 5:
+        return 80
+    else:
+        return 120
+    
+@spaces.GPU(duration=get_duration)
 def generate_video(
     input_image,
     prompt: str,