Commit ·
15a1a9a
1
Parent(s): 3072d42
Include BiRefNet models in deployment
Browse files- download_models.py +4 -1
download_models.py
CHANGED
|
@@ -2,5 +2,8 @@ from rembg import new_session
|
|
| 2 |
|
| 3 |
if __name__ == "__main__":
|
| 4 |
print("Pre-downloading rembg models...")
|
| 5 |
-
new_session("
|
|
|
|
|
|
|
|
|
|
| 6 |
print("Download complete.")
|
|
|
|
| 2 |
|
| 3 |
if __name__ == "__main__":
|
| 4 |
print("Pre-downloading rembg models...")
|
| 5 |
+
new_session("birefnet-general")
|
| 6 |
+
new_session("birefnet-general-lite")
|
| 7 |
+
new_session("isnet-anime")
|
| 8 |
+
new_session("u2net")
|
| 9 |
print("Download complete.")
|