Update app.py

app.py

@@ -160,7 +160,7 @@ preset_options = [
     {"label": "1216x704, 41 frames", "width": 1216, "height": 704, "num_frames": 41},
     {"label": "1088x704, 49 frames", "width": 1088, "height": 704, "num_frames": 49},
     {"label": "1056x640, 57 frames", "width": 1056, "height": 640, "num_frames": 57},
-    {"label": "448x448, 100 frames", "width": 448, "height": 448, "num_frames": 200},
+    {"label": "448x448, 100 frames", "width": 448, "height": 448, "num_frames": 100},
     {"label": "448x448, 200 frames", "width": 448, "height": 448, "num_frames": 200},
     {"label": "640x640, 80 frames", "width": 640, "height": 640, "num_frames": 80},
     {"label": "768x768, 64 frames", "width": 768, "height": 768, "num_frames": 64},
@@ -225,7 +225,19 @@ pipeline = XoraVideoPipeline(
     vae=vae,
 ).to(torch.bfloat16).to(device)
 
-@spaces.GPU(duration=80)
+GPU_DURATION_OPTIONS = {
+    "Short (45s)": 45,
+    "Short (60s)": 60,
+    "Medium (80s)": 80,
+    "Medium (100s)": 100,
+    "Long (120s)": 120,
+    "Long (140s)": 140,
+}
+
+def set_gpu_duration(duration_choice):
+    os.environ["GPU_DURATION"] = str(GPU_DURATION_OPTIONS[duration_choice])
+
+@spaces.GPU(duration=int(os.getenv("GPU_DURATION", "80")))  # Dynamic duration
 def generate_video_from_text(
     prompt="",
     enhance_prompt_toggle=False,
@@ -304,7 +316,7 @@ def generate_video_from_text(
     torch.cuda.empty_cache()
     return output_path
     
-@spaces.GPU(duration=80)
+@spaces.GPU(duration=int(os.getenv("GPU_DURATION", "80")))  # Dynamic duration
 def generate_video_from_image(
     image_path,
     prompt="",
@@ -401,6 +413,14 @@ def create_advanced_options():
         seed = gr.Slider(label="4.1 Seed", minimum=0, maximum=1000000, step=1, value=646373)
         inference_steps = gr.Slider(label="4.2 Inference Steps", minimum=1, maximum=50, step=1, value=35)
         guidance_scale = gr.Slider(label="4.3 Guidance Scale", minimum=1.0, maximum=5.0, step=0.1, value=4.2)
+        
+        gpu_duration = gr.Dropdown(
+            label="GPU Duration",
+            choices=list(GPU_DURATION_OPTIONS.keys()),
+            value="Medium (60s)"  # Default value
+        )
+        
+        gpu_duration.change(fn=set_gpu_duration, inputs=gpu_duration, outputs=[])
 
         height_slider = gr.Slider(
             label="4.4 Height",