gaur3009 commited on
Commit
732979b
·
verified ·
1 Parent(s): 4a3c571

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="u2net.pth")
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