Update app.py

app.py

@@ -142,13 +142,19 @@ def get_duration(
     randomize_seed,
     progress,
 ):
-    BASE_FRAMES_HEIGHT_WIDTH = 81 * 832 * 624
-    BASE_STEP_DURATION = 15
+    BASE_FRAMES_HEIGHT_WIDTH = 120 * 640 * 480
+    BASE_STEP_DURATION = 10
+    
     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
+    step_duration = BASE_STEP_DURATION * factor
+
+    estimated_duration = 10 + int(steps) * step_duration
+    MAX_ALLOWED_DURATION = 120
+    
+    return min(estimated_duration, MAX_ALLOWED_DURATION)
 
 @spaces.GPU(duration=get_duration)
 def generate_video(