Update app.py

app.py

@@ -27,12 +27,12 @@ MULTIPLE_OF = 16
 MAX_SEED = np.iinfo(np.int32).max
 
 FIXED_FPS = 16
-MIN_FRAMES_MODEL = 9
-MAX_FRAMES_MODEL = 121
+MIN_FRAMES_MODEL = 8
+MAX_FRAMES_MODEL = 120
 
+DEFAULT_DURATION = 5.0
 MIN_DURATION = round(MIN_FRAMES_MODEL/FIXED_FPS,1)
 MAX_DURATION = round(MAX_FRAMES_MODEL/FIXED_FPS,1)
-DEFAULT_DURATION = 5.1
 
 
 pipe = WanImageToVideoPipeline.from_pretrained(MODEL_ID,
@@ -123,6 +123,14 @@ def resize_image(image: Image.Image) -> Image.Image:
     return image_to_resize.resize((final_w, final_h), Image.LANCZOS)
 
 
+def get_num_frames(duration_seconds: float):
+    return 1 + int(np.clip(
+        int(round(duration_seconds * FIXED_FPS)),
+        MIN_FRAMES_MODEL,
+        MAX_FRAMES_MODEL,
+    ))
+
+
 def get_duration(
     input_image,
     prompt,
@@ -135,7 +143,13 @@ def get_duration(
     randomize_seed,
     progress,
 ):
-    return int(steps) * 15
+    BASE_FRAMES_HEIGHT_WIDTH = 81 * 832 * 624
+    BASE_STEP_DURATION = 15
+    width, height = 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
 
 @spaces.GPU(duration=get_duration)
 def generate_video(
@@ -193,7 +207,7 @@ def generate_video(
     if input_image is None:
         raise gr.Error("Please upload an input image.")
     
-    num_frames = np.clip(int(round(duration_seconds * FIXED_FPS)), MIN_FRAMES_MODEL, MAX_FRAMES_MODEL)
+    num_frames = get_num_frames(duration_seconds)
     current_seed = random.randint(0, MAX_SEED) if randomize_seed else int(seed)
     resized_image = resize_image(input_image)