Update app.py

app.py

@@ -80,14 +80,14 @@ def extract_exif_data(image):
     # and return the exif bytes. For now, it does nothing.
     return None
 
-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
+# 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
 
 
 class UpscalerApp:
@@ -304,7 +304,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)
 
-    @on_gpu_configurable(duration=59)
+    # @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,7 +369,7 @@ class UpscalerApp:
                         output_image_playground = gr.Image(label="Upscaled Result", interactive=False)
 
                 run_button_playground.click(
-                    fn=self.playground_upscale,
+                    fn=gpu_tab2,
                     inputs=[input_image_playground, upscaler_model_dropdown, upscaler_size_slider, tile_slider, tile_overlap_slider, half_checkbox],
                     outputs=[output_image_playground]
                 )
@@ -380,9 +380,9 @@ class UpscalerApp:
         self.ui.launch(**kwargs)
 
 
-@spaces.GPU
-def dummy_gpu():
-    return None
+@spaces.GPU(duration=60)
+def gpu_tab2(*args, **kwargs):
+    return self.playground_upscale(*args, **kwargs)
 
 # --- Main Execution Block ---
 if __name__ == "__main__":