Spaces:
Sleeping
Sleeping
Amit Kumar
commited on
Commit
·
d24f85a
1
Parent(s):
b4f8087
Update model trained on the whole Food101 dataset
Browse files
.gitattributes
CHANGED
@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
pretrained_effnetb2_feature_extractor_food101_20_percent.pth filter=lfs diff=lfs merge=lfs -text
|
|
|
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
pretrained_effnetb2_feature_extractor_food101_20_percent.pth filter=lfs diff=lfs merge=lfs -text
|
37 |
+
pretrained_effnetb2_feature_extractor_food101.pth filter=lfs diff=lfs merge=lfs -text
|
app.py
CHANGED
@@ -14,7 +14,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(
|
|
14 |
)
|
15 |
|
16 |
effnetb2.load_state_dict(
|
17 |
-
torch.load(f="
|
18 |
map_location=torch.device("cpu")) # Load the model to the CPU
|
19 |
)
|
20 |
|
|
|
14 |
)
|
15 |
|
16 |
effnetb2.load_state_dict(
|
17 |
+
torch.load(f="pretrained_effnetb2_feature_extractor_food101.pth",
|
18 |
map_location=torch.device("cpu")) # Load the model to the CPU
|
19 |
)
|
20 |
|
pretrained_effnetb2_feature_extractor_food101_20_percent.pth → pretrained_effnetb2_feature_extractor_food101.pth
RENAMED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e05de434aa004ffcb2d82bd1fca54a2628e8fdcda893c6ec4e0332e54fd91b49
|
3 |
+
size 31849018
|