bluspater commited on
Commit
a016ce7
·
verified ·
1 Parent(s): 3b9368e

Update u2net_utils.py

Browse files
Files changed (1) hide show
  1. u2net_utils.py +1 -1
u2net_utils.py CHANGED
@@ -9,7 +9,7 @@ from model.u2net import U2NET
9
 
10
  MODEL_DIR = "model"
11
  MODEL_NAME = "u2net.pth"
12
- MODEL_PATH = os.path.join(MODEL_DIR, MODEL_NAME)
13
  MODEL_URL = "https://huggingface.co/blurred-png/u2net-pth/resolve/main/u2net.pth"
14
 
15
  def download_model():
 
9
 
10
  MODEL_DIR = "model"
11
  MODEL_NAME = "u2net.pth"
12
+ MODEL_PATH = 'model/u2net.pth'
13
  MODEL_URL = "https://huggingface.co/blurred-png/u2net-pth/resolve/main/u2net.pth"
14
 
15
  def download_model():