Back to original max duration

app.py

@@ -28,9 +28,8 @@ MAX_SEED = np.iinfo(np.int32).max
 
 FIXED_FPS = 16
 MIN_FRAMES_MODEL = 8
-MAX_FRAMES_MODEL = 120
+MAX_FRAMES_MODEL = 80
 
-DEFAULT_DURATION = 5.0
 MIN_DURATION = round(MIN_FRAMES_MODEL/FIXED_FPS,1)
 MAX_DURATION = round(MAX_FRAMES_MODEL/FIXED_FPS,1)
 
@@ -145,7 +144,7 @@ def get_duration(
 ):
     BASE_FRAMES_HEIGHT_WIDTH = 81 * 832 * 624
     BASE_STEP_DURATION = 15
-    width, height = input_image.size
+    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
@@ -157,7 +156,7 @@ def generate_video(
     prompt,
     steps = 4,
     negative_prompt=default_negative_prompt,
-    duration_seconds = DEFAULT_DURATION,
+    duration_seconds = MAX_DURATION,
     guidance_scale = 1,
     guidance_scale_2 = 1,    
     seed = 42,