Update app.py

app.py

@@ -30,8 +30,9 @@ import cv2
 import numpy
 import os
 import random
+import inspect
 
-from basicsr.archs.rrdbnet_arch import RRDBNet
+from basicsr.archs.rrdbnet_arch import RRDBNet as _RRDBNet
 from basicsr.utils.download_util import load_file_from_url
 
 from realesrgan import RealESRGANer
@@ -112,30 +113,78 @@ def model_tip_text(model_name: str) -> str:
     return tips.get(model_name, "")
 
 
+# ────────────────────────────────────────────────────────
+# RRDBNet builder that tolerates different Basicsr signatures
+# ────────────────────────────────────────────────────────
+def build_rrdb(scale: int, num_block: int):
+    """
+    Creates an RRDBNet across several possible constructor signatures used by basicsr/realesrgan.
+    Tries, in order:
+      1) keyword style (num_in_ch/num_out_ch/num_feat/num_block/num_grow_ch/scale)
+      2) alt keyword style (in_nc/out_nc/nf/nb/gc/sf)
+      3) positional with gc before scale
+      4) positional with scale before gc
+    """
+    # Try keyword: "num_*" + "scale"
+    try:
+        return _RRDBNet(
+            num_in_ch=3, num_out_ch=3,
+            num_feat=64, num_block=num_block,
+            num_grow_ch=32, scale=scale
+        )
+    except TypeError:
+        pass
+
+    # Try keyword: "in_nc/out_nc" + "sf"
+    try:
+        return _RRDBNet(
+            in_nc=3, out_nc=3,
+            nf=64, nb=num_block,
+            gc=32, sf=scale
+        )
+    except TypeError:
+        pass
+
+    # Inspect parameters to guess positional order
+    params = list(inspect.signature(_RRDBNet).parameters.keys())
+
+    # Common positional (gc, scale) order
+    try:
+        return _RRDBNet(3, 3, 64, num_block, 32, scale)
+    except TypeError:
+        pass
+
+    # Alternate positional (scale, gc) order
+    try:
+        return _RRDBNet(3, 3, 64, num_block, scale, 32)
+    except TypeError as e:
+        raise TypeError(f"RRDBNet signature not recognized: {e}")
+
+
 # ────────────────────────────────────────────────────────
 # Core upscaling
 # Decorated for Hugging Face Spaces ZeroGPU
 # ────────────────────────────────────────────────────────
-@GPU()  # <-- lets Spaces know this function uses GPU; safe no-op locally
+@GPU()  # lets Spaces know this function uses GPU; safe no-op locally
 def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
     if img is None:
         return
 
     # ----- Select backbone + weights -----
     if model_name == 'RealESRGAN_x4plus':
-        model = RRDBNet(3, 3, 64, 23, 32, scale=4); netscale = 4
+        model = build_rrdb(scale=4, num_block=23); netscale = 4
         file_url = ['https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth']
 
     elif model_name == 'RealESRNet_x4plus':
-        model = RRDBNet(3, 3, 64, 23, 32, scale=4); netscale = 4
+        model = build_rrdb(scale=4, num_block=23); netscale = 4
         file_url = ['https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.1/RealESRNet_x4plus.pth']
 
     elif model_name == 'RealESRGAN_x4plus_anime_6B':
-        model = RRDBNet(3, 3, 64, 6, 32, scale=4); netscale = 4
+        model = build_rrdb(scale=4, num_block=6); netscale = 4
         file_url = ['https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth']
 
     elif model_name == 'RealESRGAN_x2plus':
-        model = RRDBNet(3, 3, 64, 23, 32, scale=2); netscale = 2
+        model = build_rrdb(scale=2, num_block=23); netscale = 2
         file_url = ['https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth']
 
     elif model_name == 'realesr-general-x4v3':
@@ -283,7 +332,6 @@ def main():
         input_image.change(fn=image_properties, inputs=input_image, outputs=input_image_properties)
         model_name.change(fn=model_tip_text, inputs=model_name, outputs=model_tips)
 
-        # IMPORTANT: call the @GPU-decorated function here
         restore_btn.click(
             fn=realesrgan,
             inputs=[input_image, model_name, denoise_strength, face_enhance, outscale],
@@ -293,8 +341,8 @@ def main():
 
         gr.Markdown("")  # spacer
 
-    # Disable SSR as suggested by the log
-    demo.launch(ssr_mode=False)  # share=True if you want a public link
+    # Disable SSR (ZeroGPU + Gradio logs suggested turning this off)
+    demo.launch(ssr_mode=False)  # set share=True for a public link
 
 
 if __name__ == "__main__":