Update app.py

app.py

@@ -573,7 +573,26 @@ def on_highres_toggle(first_image, last_image, high_res):
     return gr.update(value=w), gr.update(value=h)
 
 
-@spaces.GPU(duration=105)
+def get_gpu_duration(
+    first_image,
+    last_image,
+    input_audio,
+    prompt: str,
+    duration: float,
+    gpu_duration: float,
+    enhance_prompt: bool = True,
+    seed: int = 42,
+    randomize_seed: bool = True,
+    height: int = 1024,
+    width: int = 1536,
+    pose_strength: float = 0.0,
+    general_strength: float = 0.0,
+    motion_strength: float = 0.0,
+    progress=None,
+):
+    return int(gpu_duration)
+
+@spaces.GPU(duration=get_gpu_duration)
 @torch.inference_mode()
 def generate_video(
     first_image,
@@ -581,6 +600,7 @@ def generate_video(
     input_audio,
     prompt: str,
     duration: float,
+    gpu_duration: float,
     enhance_prompt: bool = True,
     seed: int = 42,
     randomize_seed: bool = True,
@@ -712,6 +732,13 @@ with gr.Blocks(title="LTX-2.3 Heretic Distilled") as demo:
 
         with gr.Column():
             output_video = gr.Video(label="Generated Video", autoplay=False)
+            gpu_duration = gr.Slider(
+                label="ZeroGPU duration (seconds)",
+                minimum=40.0,
+                maximum=240.0,
+                value=85.0,
+                step=1.0,
+            )
 
     gr.Examples(
         examples=[
@@ -726,6 +753,7 @@ with gr.Blocks(title="LTX-2.3 Heretic Distilled") as demo:
                 "crowd together closely, forming a symmetrical cluster while staring "
                 "directly into the lens.",
                 3.0,
+                80.0,
                 False,
                 42,
                 True,
@@ -737,7 +765,7 @@ with gr.Blocks(title="LTX-2.3 Heretic Distilled") as demo:
             ],
         ],
         inputs=[
-            first_image, last_image, input_audio, prompt, duration,
+            first_image, last_image, input_audio, prompt, duration, gpu_duration,
             enhance_prompt, seed, randomize_seed, height, width,
             pose_strength, general_strength, motion_strength,
         ],
@@ -764,7 +792,7 @@ with gr.Blocks(title="LTX-2.3 Heretic Distilled") as demo:
     generate_btn.click(
         fn=generate_video,
         inputs=[
-            first_image, last_image, input_audio, prompt, duration, enhance_prompt,
+            first_image, last_image, input_audio, prompt, duration, gpu_duration, enhance_prompt,
             seed, randomize_seed, height, width,
             pose_strength, general_strength, motion_strength,
         ],