Enhance GPU decorator to accept duration parameter for better control

app.py

@@ -23,11 +23,16 @@ else:
 
     class spaces:
         @staticmethod
-        def GPU(func):
-            def wrapper(*args, **kwargs):
-                return func(*args, **kwargs)
+        def GPU(func=None, duration=300):
+            def decorator(f):
+                def wrapper(*args, **kwargs):
+                    return f(*args, **kwargs)
 
-            return wrapper
+                return wrapper
+
+            if func is None:
+                return decorator
+            return decorator(func)
 
 
 # Define constants
@@ -252,7 +257,7 @@ with gr.Blocks() as demo:
 
         return current_model, current_processor
 
-    @spaces.GPU
+    @spaces.GPU(duration=300)
     def generate(
         model,
         processor,