Update app.py

app.py

@@ -162,10 +162,10 @@ GPU_DURATION_OPTIONS = {
     "Long (140s)": 140,
 }
 
-# Initialize Gradio state
-gpu_duration_state = gr.State(60)  # Initial value
+# Global variable to store GPU duration
+global_gpu_duration = 60  # Initial value
 
-@spaces.GPU(duration=gpu_duration_state.value)  # Access state value
+@spaces.GPU(duration=global_gpu_duration)
 def generate(
     model_choice: str,
     prompt: str,
@@ -270,7 +270,8 @@ def generate_cpu(
     return image_paths, seed
 
 def set_gpu_duration(duration_choice):
-    return GPU_DURATION_OPTIONS[duration_choice]  # Update state
+    global global_gpu_duration
+    global_gpu_duration = GPU_DURATION_OPTIONS[duration_choice]
 
 def load_predefined_images1():
     predefined_images1 = [
@@ -324,19 +325,13 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
         )
 
     with gr.Accordion("Advanced options", open=False, visible=True):
-        duration_dropdown = gr.Dropdown(
-            choices=[
-            "Short (25s)",
-            "Short (45s)",
-            "Medium (60s)",
-            "Medium (80s)",
-            "Long (100s)",
-            "Long (120s)",
-            "Long (140s)",
-            ]
+        gpu_duration = gr.Dropdown(
+            label="GPU Duration",
+            choices=list(GPU_DURATION_OPTIONS.keys()),
+            value="Medium (60s)"  # Default value
         )
         
-        duration_dropdown.change(set_gpu_duration, inputs=duration_dropdown, outputs=gpu_duration_state)
+        gpu_duration.change(fn=set_gpu_duration, inputs=gpu_duration, outputs=[])
 
         style_selection = gr.Radio(
             show_label=True,