Spaces:
Runtime error
Runtime error
Alex Strick van Linschoten
commited on
Commit
·
3340a5b
1
Parent(s):
77b425a
add better model
Browse files- app.py +2 -1
- 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
|