Update app.py

app.py

@@ -141,16 +141,30 @@ def get_num_frames(duration_seconds: float):
         MAX_FRAMES_MODEL,
     ))
 
-def get_duration(input_image, prompt, steps, negative_prompt, duration_seconds, guidance_scale, guidance_scale_2, seed, randomize_seed, progress):
+def get_duration(
+    input_image,
+    prompt,
+    steps,
+    negative_prompt,
+    duration_seconds,
+    guidance_scale,
+    guidance_scale_2,
+    seed,
+    randomize_seed,
+    progress,
+):
     BASE_FRAMES_HEIGHT_WIDTH = 81 * 832 * 624
     BASE_STEP_DURATION = 15
     width, height = resize_image(input_image).size
     frames = get_num_frames(duration_seconds)
     factor = frames * width * height / BASE_FRAMES_HEIGHT_WIDTH
     step_duration = BASE_STEP_DURATION * factor ** 1.5
-    return 10 + int(steps) * step_duration
+    est = 10 + int(steps) * step_duration
+    
+    MAX_ALLOWED = 30
+    return min(est, MAX_ALLOWED)
 
-@spaces.GPU(duration=get_duration)
+@spaces.GPU
 def generate_video(
     input_image,
     prompt,