Spaces:
Build error
Build error
init
Browse files
app.py
CHANGED
@@ -67,7 +67,7 @@ prediction_dir = 'portrait_results'
|
|
67 |
if(not os.path.exists(prediction_dir)):
|
68 |
os.mkdir(prediction_dir)
|
69 |
|
70 |
-
model_dir = os.path.
|
71 |
|
72 |
|
73 |
# --------- 3. model define ---------
|
|
|
67 |
if(not os.path.exists(prediction_dir)):
|
68 |
os.mkdir(prediction_dir)
|
69 |
|
70 |
+
model_dir = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'U-2-Net/saved_models/u2net_portrait/u2net_portrait.pth')
|
71 |
|
72 |
|
73 |
# --------- 3. model define ---------
|