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="pretrained_effnetb2_feature_extractor_food101_20_percent.pth",
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:11d2f79bfdeb19d7cb1dc41a2a98e120dfc84ea339e03a166c9c5ce9910886fc
3
- size 31855674
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e05de434aa004ffcb2d82bd1fca54a2628e8fdcda893c6ec4e0332e54fd91b49
3
+ size 31849018