Allocation time

app.py

@@ -193,7 +193,13 @@ def generate_video(
     randomize_seed=False,
     progress=gr.Progress(track_tqdm=True)
 ):
-    return generate_video(
+    if input_image_debug_value[0] is not None or end_image_debug_value[0] is not None or prompt_debug_value[0] is not None or total_second_length_debug_value[0] is not None:
+        start_image_pil = input_image_debug_value[0]
+        end_image_pil = end_image_debug_value[0]
+        prompt = prompt_debug_value[0]
+        duration_seconds = min(total_second_length_debug_value[0] * 60 * 100, 600)
+
+    return generate_video_on_gpu(
     start_image_pil,
     end_image_pil,
     prompt,
@@ -207,31 +213,39 @@ def generate_video(
     progress
 )
 
-@spaces.GPU(duration=120)
+def get_duration(
+    start_image_pil,
+    end_image_pil,
+    prompt,
+    negative_prompt,
+    duration_seconds,
+    steps,
+    guidance_scale,
+    guidance_scale_2,
+    seed,
+    randomize_seed,
+    progress
+):
+    return duration_seconds
+
+@spaces.GPU(duration=get_duration)
 def generate_video_on_gpu(
     start_image_pil,
     end_image_pil,
     prompt,
-    negative_prompt=default_negative_prompt,
-    duration_seconds=2.1,
-    steps=8,
-    guidance_scale=1,
-    guidance_scale_2=1,
-    seed=42,
-    randomize_seed=False,
-    progress=gr.Progress(track_tqdm=True)
+    negative_prompt,
+    duration_seconds,
+    steps,
+    guidance_scale,
+    guidance_scale_2,
+    seed,
+    randomize_seed,
+    progress
 ):
     """
     Generates a video by interpolating between a start and end image, guided by a text prompt,
     using the diffusers Wan2.2 pipeline.
     """
-    if input_image_debug_value[0] is not None or end_image_debug_value[0] is not None or prompt_debug_value[0] is not None or total_second_length_debug_value[0] is not None:
-        start_image_pil = input_image_debug_value[0]
-        end_image_pil = end_image_debug_value[0]
-        prompt = prompt_debug_value[0]
-        duration_seconds = total_second_length_debug_value[0]
-        #allocation_time = min(total_second_length_debug_value[0] * 60 * 100, 600)
-
     if start_image_pil is None or end_image_pil is None:
         raise gr.Error("Please upload both a start and an end image.")