Update app.py

app.py

@@ -1,19 +1,28 @@
 import gradio as gr
-import spaces
 import cv2
 import numpy
 import os
 import random
+import torch
 from basicsr.archs.rrdbnet_arch import RRDBNet
 from basicsr.utils.download_util import load_file_from_url
 from realesrgan import RealESRGANer
 from realesrgan.archs.srvgg_arch import SRVGGNetCompact
 from torchvision.transforms.functional import rgb_to_grayscale
 
+# Enable HF Spaces GPU wrapper only when explicitly requested
+# Set env USE_SPACES_GPU=1 on Hugging Face Spaces to enable GPU decorator
+def maybe_spaces_gpu(fn):
+    use_gpu_decorator = os.getenv("USE_SPACES_GPU", "0") == "1"
+    if use_gpu_decorator:
+        import spaces
+        return spaces.GPU(duration=120)(fn)
+    return fn
+
 last_file = None
 img_mode = "RGBA"
 
-@spaces.GPU(duration=120)
+@maybe_spaces_gpu
 def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
     """Real-ESRGAN function to restore (and upscale) images.
     """
@@ -62,16 +71,17 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
         dni_weight = [denoise_strength, 1 - denoise_strength]
 
     # Restorer Class
+    use_cuda = torch.cuda.is_available()
     upsampler = RealESRGANer(
         scale=netscale,
         model_path=model_path,
         dni_weight=dni_weight,
         model=model,
-        tile=0,
+        tile=128,
         tile_pad=10,
         pre_pad=10,
-        half=False,
-        gpu_id=None
+        half=use_cuda,
+        gpu_id=0 if use_cuda else None
     )
 
     # Use GFPGAN for face enhancement
@@ -167,7 +177,7 @@ def image_properties(img):
 
 def main():
     # Gradio Interface
-    with gr.Blocks(title="Real-ESRGAN Gradio Demo", theme="dark") as demo:
+    with gr.Blocks(title="Real-ESRGAN Gradio Demo", theme=gr.themes.Soft()) as demo:
 
         gr.Markdown(
             """# <div align="center"> Ilaria Upscaler 💖 </div>