Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ from io import BytesIO
|
|
14 |
from huggingface_hub import hf_hub_download
|
15 |
|
16 |
# Download u2net.pth from HuggingFace Hub
|
17 |
-
model_path = hf_hub_download(repo_id="BritishWerewolf/U-2-Net", filename="
|
18 |
|
19 |
# Use a known U2NET implementation (e.g., from https://github.com/xuebinqin/U-2-Net/blob/master/u2net_test.py)
|
20 |
from u2net import U2NET # Assume you copied the model code as u2net.py
|
|
|
14 |
from huggingface_hub import hf_hub_download
|
15 |
|
16 |
# Download u2net.pth from HuggingFace Hub
|
17 |
+
model_path = hf_hub_download(repo_id="BritishWerewolf/U-2-Net", filename="onnx/model.onnx")
|
18 |
|
19 |
# Use a known U2NET implementation (e.g., from https://github.com/xuebinqin/U-2-Net/blob/master/u2net_test.py)
|
20 |
from u2net import U2NET # Assume you copied the model code as u2net.py
|