Update app.py

app.py

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