Fix gradio version, model repo, and remove rmbg-v2

- Bump gradio to 5.47.0 and pin all deps from reference project
- Update sdk_version to 5.47.0 in README
- Switch HF_REPO_ID to tellurion/sdxl
- Fetch available models dynamically via list_repo_files()
- Change default_mask_extractor to ISNet (non-gated)
- Remove rmbg-v2 (gated briaai/RMBG-2.0 repo, inaccessible without token)

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

README.md

@@ -4,7 +4,7 @@ emoji: 🎨
 colorFrom: purple
 colorTo: pink
 sdk: gradio
-sdk_version: "4.44.1"
+sdk_version: "5.47.0"
 python_version: "3.10"
 app_file: app.py
 pinned: false

backend/__init__.py

@@ -11,6 +11,6 @@ __all__ = [
 
 
 default_line_extractor = "lineart_keras"
-default_mask_extractor = "rmbg-v2"
-mask_extractor_list = ["none", "ISNet", "rmbg-v2", "BiRefNet", "BiRefNet_HR"]
+default_mask_extractor = "ISNet"
+mask_extractor_list = ["none", "ISNet", "BiRefNet", "BiRefNet_HR"]
 line_extractor_list = ["lineart", "lineart_denoise", "lineart_keras", "lineart_sk"]

backend/appfunc.py

@@ -4,7 +4,7 @@ import traceback
 import gradio as gr
 import os.path as osp
 
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, list_repo_files
 
 from omegaconf import OmegaConf
 from refnet.util import instantiate_from_config
@@ -22,15 +22,9 @@ smask_extractor = create_model("ISNet-sketch").cpu()
 MAXM_INT32 = 429496729
 
 # HuggingFace model repository
-HF_REPO_ID = "tellurion/colorizer"
+HF_REPO_ID = "tellurion/sdxl"
 MODEL_CACHE_DIR = "models"
 
-# Model registry: filename -> model_type
-MODEL_REGISTRY = {
-    "sdxl.safetensors": "sdxl",
-    "xlv2.safetensors": "xlv2",
-}
-
 model_types = ["sdxl", "xlv2"]
 
 '''
@@ -39,8 +33,12 @@ model_types = ["sdxl", "xlv2"]
 
 
 def get_available_models():
-    """Return list of available model names from registry."""
-    return list(MODEL_REGISTRY.keys())
+    """Fetch available .safetensors files from HuggingFace Hub."""
+    try:
+        files = list_repo_files(HF_REPO_ID)
+        return [f for f in files if f.endswith(".safetensors")]
+    except Exception:
+        return []
 
 
 def download_model(filename):
@@ -104,13 +102,12 @@ def load_model(ckpt_name):
     config_root = "configs/inference"
 
     try:
-        # Determine model type from registry or filename prefix
-        new_model_type = MODEL_REGISTRY.get(ckpt_name, "")
-        if not new_model_type:
-            for key in model_types:
-                if ckpt_name.startswith(key):
-                    new_model_type = key
-                    break
+        # Determine model type from filename prefix
+        new_model_type = ""
+        for key in model_types:
+            if ckpt_name.startswith(key):
+                new_model_type = key
+                break
 
         if model_type != new_model_type or not "model" in globals():
             if "model" in globals() and exists(model):

preprocessor/__init__.py

@@ -43,7 +43,6 @@ remote_model_dict = {
 }
 
 BiRefNet_dict = {
-    "rmbg-v2": ("briaai/RMBG-2.0", 1024),
     "BiRefNet": ("ZhengPeng7/BiRefNet", 1024),
     "BiRefNet_HR": ("ZhengPeng7/BiRefNet_HR", 2048)
 }

requirements.txt

@@ -1,19 +1,21 @@
-accelerate
-diffusers
-einops
-huggingface-hub<0.29.0
-kornia
-numpy
-omegaconf
-open-clip-torch
+accelerate==1.10.1
+diffusers==0.35.1
+einops==0.8.1
+gradio==5.47.0
+gradio-client==1.13.2
+huggingface-hub==0.35.1
+kornia==0.8.1
+numpy==2.2.6
+omegaconf==2.3.0
+open-clip-torch==2.24.0
 opencv-python-headless
-pillow
-safetensors
-timm
-tokenizers
-torch
-torchdiffeq
-torchsde
-torchvision
-tqdm
-transformers
+pillow==11.3.0
+safetensors==0.6.2
+timm==1.0.20
+tokenizers==0.22.1
+torch==2.8.0
+torchdiffeq==0.2.5
+torchsde==0.2.6
+torchvision==0.23.0
+tqdm==4.67.1
+transformers==4.57.0