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 ---------
|