Update app.py

app.py

@@ -165,7 +165,16 @@ GPU_DURATION_OPTIONS = {
 # Global variable to store GPU duration
 global_gpu_duration = 60  # Initial value
 
-@spaces.GPU(duration=global_gpu_duration)
+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 generate(
     model_choice: str,
     prompt: str,