Update app.py

app.py

@@ -162,19 +162,25 @@ GPU_DURATION_OPTIONS = {
     "Long (140s)": 140,
 }
 
-# Global variable to store GPU duration
-global_gpu_duration = 60  # Initial value
-
-def gpu_with_dynamic_duration(func):
-    def wrapper(*args, **kwargs):
-        # Apply @spaces.GPU with the current global_gpu_duration
-        @spaces.GPU(duration=global_gpu_duration)
-        def inner(*args, **kwargs):
-            return func(*args, **kwargs)
-        return inner(*args, **kwargs)
-    return wrapper
-
-@gpu_with_dynamic_duration
+def set_gpu_duration(duration_choice):
+    os.environ["GPU_DURATION"] = str(GPU_DURATION_OPTIONS[duration_choice])
+
+@gpu_with_duration(duration=45)
+def generate_45s(*args, **kwargs):
+    return generate(*args, **kwargs)  # Call the common generate function
+
+@gpu_with_duration(duration=60)
+def generate_60s(*args, **kwargs):
+    return generate(*args, **kwargs)
+
+@gpu_with_duration(duration=80)
+def generate_80s(*args, **kwargs):
+    return generate(*args, **kwargs)
+
+@gpu_with_duration(duration=100)
+def generate_100s(*args, **kwargs):
+    return generate(*args, **kwargs)
+
 def generate(
     model_choice: str,
     prompt: str,
@@ -322,8 +328,11 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
             placeholder="Enter your prompt",
             container=False,
         )
-        run_button = gr.Button("Run", scale=0)
+        run_button_45 = gr.Button("Run_45", scale=0)
         cpu_run_button = gr.Button("CPU Run", scale=0)
+        run_button_60 = gr.Button("Run_60", scale=0)
+        run_button_80 = gr.Button("Run_80", scale=0)
+        run_button_100 = gr.Button("Run_100", scale=0)
     result = gr.Gallery(label="Result", columns=1, show_label=False) 
 
     with gr.Row():
@@ -422,12 +431,56 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
     
     gr.on(
         triggers=[
-            prompt.submit,
-            negative_prompt.submit,
-            run_button.click,
+            run_button_100.click,
+        ],
+      #  api_name="generate",  # Add this line
+        fn=generate_100s,
+        inputs=[
+            model_choice,
+            prompt,
+            negative_prompt,
+            use_negative_prompt,
+            style_selection,
+            seed,
+            width,
+            height,
+            guidance_scale,
+            num_inference_steps,
+            randomize_seed, 
+            num_images, 
+        ],
+        outputs=[result, seed],
+    )
+    
+    gr.on(
+        triggers=[
+            run_button_80.click,
+        ],
+      #  api_name="generate",  # Add this line
+        fn=generate_80s,
+        inputs=[
+            model_choice,
+            prompt,
+            negative_prompt,
+            use_negative_prompt,
+            style_selection,
+            seed,
+            width,
+            height,
+            guidance_scale,
+            num_inference_steps,
+            randomize_seed, 
+            num_images, 
+        ],
+        outputs=[result, seed],
+    )
+    
+    gr.on(
+        triggers=[
+            run_button_60.click,
         ],
-        api_name="generate",  # Add this line
-        fn=generate,
+      #  api_name="generate",  # Add this line
+        fn=generate_60s,
         inputs=[
             model_choice,
             prompt,
@@ -447,10 +500,10 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
 
     gr.on(
         triggers=[
-            cpu_run_button.click,
+            run_button_45.click,
         ],
-        api_name="generate",  # Add this line
-        fn=generate_cpu,
+      #  api_name="generate",  # Add this line
+        fn=generate_45s,
         inputs=[
             model_choice,
             prompt,