Update app_fast.py

app_fast.py

@@ -95,13 +95,13 @@ def get_duration(input_image, prompt, height, width,
                    seed, randomize_seed, 
                    progress):
     if steps > 5 and duration_seconds > 5:
-        return 50
+        return 60
     elif steps > 5 or duration_seconds > 5:
-        return 45
+        return 50
     else:
         return 40
 
-@spaces.GPU
+@spaces.GPU(duration=get_duration)
 def generate_video(input_image, prompt, height, width, negative_prompt=default_negative_prompt, duration_seconds=2, guidance_scale=0, steps=4, seed=44, randomize_seed=False, progress=gr.Progress(track_tqdm=True)):
     target_h = max(MOD_VALUE, (int(height) // MOD_VALUE) * MOD_VALUE)
     target_w = max(MOD_VALUE, (int(width) // MOD_VALUE) * MOD_VALUE)