Commit ·
a2fe5a2
1
Parent(s): 6dd9966
Add inference endpoint feature in HF model page.
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -75,7 +75,7 @@ usage_to_weights_file = {
|
|
| 75 |
}
|
| 76 |
|
| 77 |
# Choose the version of BiRefNet here.
|
| 78 |
-
usage = 'General'
|
| 79 |
|
| 80 |
# Set resolution
|
| 81 |
if usage in ['General-Lite-2K']:
|
|
|
|
| 75 |
}
|
| 76 |
|
| 77 |
# Choose the version of BiRefNet here.
|
| 78 |
+
usage = 'General-legacy'
|
| 79 |
|
| 80 |
# Set resolution
|
| 81 |
if usage in ['General-Lite-2K']:
|