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

Files changed (1) hide show
  1. 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="models/",
 
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