Update app.py

app.py

@@ -100,55 +100,48 @@ def render_progress(pct: float, label: str = "") -> str:
 <div style="font-size:12px;opacity:.8;margin-top:4px;">{label} {pct:.1f}%</div>'''
 
 def build_rrdb(scale: int, num_block: int):
-    return _RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64,
-                    num_block=num_block, num_grow_ch=32, scale=scale)
+    return RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64,
+                   num_block=num_block, num_grow_ch=32, scale=scale)
 
-def _weights_dir() -> str:
-    wdir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "weights")
-    os.makedirs(wdir, exist_ok=True)
-    return wdir
-
-# ─────────────────────────────────────────────
-# Real-ESRGAN (CPU only)
-# ─────────────────────────────────────────────
-def get_realesrganer(model_id: str, scale: int, tile: int, half: bool, device: str = "cpu") -> RealESRGANer:
+def get_realesrganer(model_id: str, scale: int, tile: int,
+                     half: bool, device: str = "cpu") -> RealESRGANer:
     wdir = _weights_dir()
+
     if model_id == "x4plus":
-        model = build_rrdb(scale=4, num_block=23); netscale = 4
-        urls = ["https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth"]
+        model = build_rrdb(scale=4, num_block=23)
+        netscale = 4
+        url = "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth"
         model_path = os.path.join(wdir, "RealESRGAN_x4plus.pth")
-        dni_weight = None
     elif model_id == "x4plus-anime":
-        model = build_rrdb(scale=4, num_block=6); netscale = 4
-        urls = ["https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth"]
+        model = build_rrdb(scale=4, num_block=6)
+        netscale = 4
+        url = "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth"
         model_path = os.path.join(wdir, "RealESRGAN_x4plus_anime_6B.pth")
-        dni_weight = None
     elif model_id == "x2plus":
-        model = build_rrdb(scale=2, num_block=23); netscale = 2
-        urls = ["https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth"]
+        model = build_rrdb(scale=2, num_block=23)
+        netscale = 2
+        url = "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth"
         model_path = os.path.join(wdir, "RealESRGAN_x2plus.pth")
-        dni_weight = None
     else:
         raise ValueError(f"Unknown model_id: {model_id}")
 
-    for url in urls:
-        fname = os.path.basename(url)
-        if not os.path.isfile(os.path.join(wdir, fname)):
-            load_file_from_url(url=url, model_dir=wdir, progress=True)
+    if not os.path.isfile(model_path):
+        load_file_from_url(url=url, model_dir=wdir, progress=True)
+
+    # auto device
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    half = half and device == "cuda"
 
-    # 🔒 Force CPU only
-    upsampler = RealESRGANer(
+    return RealESRGANer(
         scale=netscale,
         model_path=model_path,
-        dni_weight=dni_weight,
         model=model,
-        tile=tile or 256,
+        tile=tile,
         tile_pad=10,
-        pre_pad=10,
-        half=False,   # CPU cannot use half precision
-        gpu_id=None   # disable GPU completely
+        pre_pad=0,
+        half=half,
+        device=device,
     )
-    return upsampler
 
 # ─────────────────────────────────────────────
 # Step 2: Sources + Processing (batch click)