Update app.py

app.py

@@ -36,7 +36,7 @@ SLIDER_MIN_H, SLIDER_MAX_H = 128, 896
 SLIDER_MIN_W, SLIDER_MAX_W = 128, 896
 MAX_SEED = np.iinfo(np.int32).max
 
-FIXED_FPS = 24
+FIXED_FPS = 60
 MIN_FRAMES_MODEL = 8
 MAX_FRAMES_MODEL = 81 
 
@@ -79,17 +79,16 @@ def handle_image_upload_for_dims_wan(uploaded_pil_image, current_h_val, current_
         gr.Warning("Error attempting to calculate new dimensions")
         return gr.update(value=DEFAULT_H_SLIDER_VALUE), gr.update(value=DEFAULT_W_SLIDER_VALUE)
 
+# --- Эта функция изменена: убрано ограничение на 3 секунды ---
 def get_duration(input_image, prompt, height, width, 
-                   negative_prompt, duration_seconds,
-                   guidance_scale, steps,
-                   seed, randomize_seed, 
-                   progress):
-    if steps > 4 and duration_seconds > 2:
-        return 90
-    elif steps > 4 or duration_seconds > 2:
-        return 75
-    else:
-        return 60
+                 negative_prompt, duration_seconds,
+                 guidance_scale, steps,
+                 seed, randomize_seed, 
+                 progress):
+    # Простая оценка времени выполнения: шаги × кадры × коэффициент
+    num_frames = int(round(duration_seconds * FIXED_FPS))
+    estimated_time = steps * num_frames * 0.2  # ~0.2 секунды на кадр/шаг
+    return max(10, int(estimated_time))  # Минимум 10 секунд для стабильности
 
 @spaces.GPU(duration=get_duration)
 def generate_video(input_image, prompt, height, width, 
@@ -125,7 +124,7 @@ def generate_video(input_image, prompt, height, width,
 
 with gr.Blocks() as demo:
     gr.Markdown("# Fast 4 steps Wan 2.1 I2V (14B) with CausVid LoRA")
-    gr.Markdown("[CausVid](https://github.com/tianweiy/CausVid) is a distilled version of Wan 2.1 to run faster in just 4-8 steps, [extracted as LoRA by Kijai](https://huggingface.co/Kijai/WanVideo_comfy/blob/main/Wan21_CausVid_14B_T2V_lora_rank32.safetensors) and is compatible with 🧨 diffusers")
+    gr.Markdown("[CausVid](https://github.com/tianweiy/CausVid   ) is a distilled version of Wan 2.1 to run faster in just 4-8 steps, [extracted as LoRA by Kijai](https://huggingface.co/Kijai/WanVideo_comfy/blob/main/Wan21_CausVid_14B_T2V_lora_rank32.safetensors   ) and is compatible with 🧨 diffusers")
     with gr.Row():
         with gr.Column():
             input_image_component = gr.Image(type="pil", label="Input Image (auto-resized to target H/W)")