Update app.py

app.py

@@ -415,6 +415,7 @@ def get_inference_duration(
     frame_multiplier,
     quality,
     duration_seconds,
+    safe_mode,
     progress
 ):
     BASE_FRAMES_HEIGHT_WIDTH = 81 * 832 * 624
@@ -425,7 +426,7 @@ def get_inference_duration(
     gen_time = int(steps) * step_duration
 
     if guidance_scale > 1:
-        gen_time = gen_time * 1.8
+        gen_time = gen_time * 1.9
 
     frame_factor = frame_multiplier // FIXED_FPS
     if frame_factor > 1:
@@ -433,7 +434,11 @@ def get_inference_duration(
         inter_time = (total_out_frames * 0.02)
         gen_time += inter_time
 
-    return 15 + gen_time
+    total_time = 15 + gen_time
+    if safe_mode:
+        total_time = total_time * 1.20
+
+    return total_time
 
 
 @spaces.GPU(duration=get_inference_duration)
@@ -452,6 +457,7 @@ def run_inference(
     frame_multiplier,
     quality,
     duration_seconds,
+    safe_mode=False,
     progress=gr.Progress(track_tqdm=True),
 ):
     scheduler_class = SCHEDULER_MAP.get(scheduler_name)
@@ -528,6 +534,7 @@ def generate_video(
     scheduler="UniPCMultistep",
     flow_shift=6.0,
     frame_multiplier=16,
+    safe_mode=False,
     video_component=True,
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -602,6 +609,7 @@ def generate_video(
         frame_multiplier,
         quality,
         duration_seconds,
+        safe_mode,
         progress,
     )
     print(f"GPU complete: {task_n}")
@@ -639,6 +647,11 @@ with gr.Blocks(delete_cache=(3600, 10800)) as demo:
                 label="Video Fluidity (Frames per Second)",
                 info="Extra frames will be generated using flow estimation, which estimates motion between frames to make the video smoother."
             )
+            safe_mode_checkbox = gr.Checkbox(
+                label="🛠️ Safe Mode",
+                value=False,
+                info="Requests 20% extra processing time to try to prevent unfinished tasks when the server is busy."
+            )
             with gr.Accordion("Advanced Settings", open=False):
                 last_image_component = gr.Image(type="pil", label="Last Image (Optional)", sources=["upload", "clipboard"])
                 negative_prompt_input = gr.Textbox(label="Negative Prompt", value=default_negative_prompt, info="Used if any Guidance Scale > 1.", lines=3)
@@ -681,6 +694,7 @@ with gr.Blocks(delete_cache=(3600, 10800)) as demo:
         negative_prompt_input, duration_seconds_input,
         guidance_scale_input, guidance_scale_2_input, seed_input, randomize_seed_checkbox,
         quality_slider, scheduler_dropdown, flow_shift_slider, frame_multi,
+        safe_mode_checkbox,
         play_result_video
     ]