Increase GPU duration due download times

app.py

@@ -24,7 +24,7 @@ else:
 
     class spaces:
         @staticmethod
-        def GPU(func):
+        def GPU(func, duration: int = 60):
             def wrapper(*args, **kwargs):
                 return func(*args, **kwargs)
 
@@ -145,7 +145,7 @@ with gr.Blocks() as demo:
     current_processor = None
     current_model_id = None
 
-    @spaces.GPU
+    @spaces.GPU(duration=300)
     def run(
         image,
         model_id: str,