Update app.py

app.py

@@ -294,7 +294,7 @@ class UpscalerApp:
         reveal_text = f"Thank you! Your preference for **{choice}** has been recorded.\n\n- **Image A was:** {model_a}\n- **Image B was:** {model_b}"
         return reveal_text, gr.Button(interactive=False), gr.Button(interactive=False)
 
-    @spaces.GPU()
+    @on_gpu_configurable(duration=59)
     def playground_upscale(self, image, upscaler_name, upscaler_size, tile, tile_overlap, half):
         if image is None or upscaler_name is None: return None
         return self.process_upscale(image, upscaler_name, upscaler_size, tile, tile_overlap, half)
@@ -369,6 +369,16 @@ class UpscalerApp:
     def launch(self, **kwargs):
         self.ui.launch(**kwargs)
 
+
+def on_gpu_configurable(duration=60):
+    def decorator(func):
+        @functools.wraps(func)
+        @spaces.GPU(duration=duration)
+        def wrapper(*args, **kwargs):
+            return func(*args, **kwargs)
+        return wrapper
+    return decorator
+
 @spaces.GPU
 def dummy_gpu():
     return None