Add new model BiRefNet_HR to the model list.

app.py

@@ -73,6 +73,7 @@ class ImagePreprocessor():
 
 usage_to_weights_file = {
     'General': 'BiRefNet',
+    'General-HR': 'BiRefNet_HR',
     'General-Lite': 'BiRefNet_lite',
     'General-Lite-2K': 'BiRefNet_lite-2K',
     'Matting': 'BiRefNet-matting',
@@ -104,8 +105,13 @@ def predict(images, resolution, weights_file):
     try:
         resolution = [int(int(reso)//32*32) for reso in resolution.strip().split('x')]
     except:
-        resolution = (1024, 1024) if weights_file not in ['General-Lite-2K'] else (2560, 1440)
-        print('Invalid resolution input. Automatically changed to 1024x1024 or 2K.')
+        if weights_file == 'General-HR':
+            resolution = (2048, 2048)
+        elif weights_file == 'General-Lite-2K':
+            resolution = (2560, 1440)
+        else:
+            resolution = (1024, 1024)
+        print('Invalid resolution input. Automatically changed to 1024x1024 / 2048x2048 / 2560x1440.')
 
     if isinstance(images, list):
         # For tab_batch

app_local.py

@@ -73,6 +73,7 @@ class ImagePreprocessor():
 
 usage_to_weights_file = {
     'General': 'BiRefNet',
+    'General-HR': 'BiRefNet_HR',
     'General-Lite': 'BiRefNet_lite',
     'General-Lite-2K': 'BiRefNet_lite-2K',
     'Matting': 'BiRefNet-matting',
@@ -104,8 +105,13 @@ def predict(images, resolution, weights_file):
     try:
         resolution = [int(int(reso)//32*32) for reso in resolution.strip().split('x')]
     except:
-        resolution = (1024, 1024) if weights_file not in ['General-Lite-2K'] else (2560, 1440)
-        print('Invalid resolution input. Automatically changed to 1024x1024 or 2K.')
+        if weights_file == 'General-HR':
+            resolution = (2048, 2048)
+        elif weights_file == 'General-Lite-2K':
+            resolution = (2560, 1440)
+        else:
+            resolution = (1024, 1024)
+        print('Invalid resolution input. Automatically changed to 1024x1024 / 2048x2048 / 2560x1440.')
 
     if isinstance(images, list):
         # For tab_batch

requirements.txt

@@ -1,13 +1,13 @@
-torch==2.0.1
-torchvision==0.15.2
-opencv-python==4.9.0.80
-tqdm==4.66.2
-timm==0.9.16
-prettytable==3.10.0
-scipy==1.12.0
-scikit-image==0.22.0
-kornia==0.7.1
-gradio_imageslider==0.0.18
-transformers==4.42.4
-huggingface_hub==0.23.4
+torch==2.5.1
+torchvision==0.20.1
+opencv-python
+tqdm
+timm
+prettytable
+scipy
+scikit-image
+kornia
+gradio_imageslider>=0.0.18
+transformers>=4.42.4
+huggingface_hub>=0.25
 einops