Fix time

app.py

@@ -197,7 +197,8 @@ def generate_video(
         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)
+        duration_seconds = total_second_length_debug_value[0]
+        allocation_time = min(duration_seconds * 60 * 100, 600)
 
     return generate_video_on_gpu(
     start_image_pil,
@@ -210,7 +211,8 @@ def generate_video(
     guidance_scale_2,
     seed,
     randomize_seed,
-    progress
+    progress,
+    allocation_time
 )
 
 def get_duration(
@@ -224,9 +226,10 @@ def get_duration(
     guidance_scale_2,
     seed,
     randomize_seed,
-    progress
+    progress,
+    allocation_time
 ):
-    return duration_seconds
+    return allocation_time
 
 @spaces.GPU(duration=get_duration)
 def generate_video_on_gpu(
@@ -240,7 +243,8 @@ def generate_video_on_gpu(
     guidance_scale_2,
     seed,
     randomize_seed,
-    progress
+    progress,
+    allocation_time
 ):
     """
     Generates a video by interpolating between a start and end image, guided by a text prompt,