Alex Strick van Linschoten commited on
Commit
3340a5b
·
1 Parent(s): 77b425a

add better model

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. best_model.pth +3 -0
app.py CHANGED
@@ -13,7 +13,8 @@ from icevision.all import *
13
  from icevision.models.checkpoint import *
14
  from PIL import Image as PILImage
15
 
16
- checkpoint_path = "./2022-01-15-vfnet-post-self-train.pth"
 
17
  checkpoint_and_model = model_from_checkpoint(checkpoint_path)
18
  model = checkpoint_and_model["model"]
19
  model_type = checkpoint_and_model["model_type"]
 
13
  from icevision.models.checkpoint import *
14
  from PIL import Image as PILImage
15
 
16
+ # checkpoint_path = "./2022-01-15-vfnet-post-self-train.pth"
17
+ checkpoint_path = "./best_model.pth"
18
  checkpoint_and_model = model_from_checkpoint(checkpoint_path)
19
  model = checkpoint_and_model["model"]
20
  model_type = checkpoint_and_model["model_type"]
best_model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e06ebdd8e5273470738ed3543d061b97d4a82d20574999dcdc2bbbbb76ace031
3
+ size 131191831