Spaces:
Running
Running
Abid Ali Awan
commited on
Commit
·
81cb914
1
Parent(s):
627d8bd
Fix model path configuration to prevent duplicate directory
Browse files- Set path='.' and model_dir='models' explicitly
- Prevents fastai from looking in models/models/model.pth
- Now correctly loads from ./models/model.pth
- app/app_savta.py +2 -1
app/app_savta.py
CHANGED
|
@@ -60,7 +60,8 @@ learner = unet_learner(
|
|
| 60 |
resnet34,
|
| 61 |
n_out=3,
|
| 62 |
loss_func=MSELossFlat(),
|
| 63 |
-
path="
|
|
|
|
| 64 |
)
|
| 65 |
learner.load("model")
|
| 66 |
|
|
|
|
| 60 |
resnet34,
|
| 61 |
n_out=3,
|
| 62 |
loss_func=MSELossFlat(),
|
| 63 |
+
path=".",
|
| 64 |
+
model_dir="models",
|
| 65 |
)
|
| 66 |
learner.load("model")
|
| 67 |
|