Mohammad Reza Hashempour commited on
Commit
972a2b8
1 Parent(s): cd4e870

add new model

Browse files
.gitattributes CHANGED
@@ -33,4 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip 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
- models/09_pretrained_effnetb2_feature_extractor_food101_20_percent.pth filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip 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
+ models/pretrained_effnetb2_feature_extractor_food101_20_percent.pth filter=lfs diff=lfs merge=lfs -text
37
+ models/*.pth filter=lfs diff=lfs merge=lfs -text
38
+ models/pretrained_effnetb2_feature_extractor_food101_100_percent.pth filter=lfs diff=lfs merge=lfs -text
app.py CHANGED
@@ -21,7 +21,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(
21
  # Load saved weights
22
  effnetb2.load_state_dict(
23
  torch.load(
24
- f="models/09_pretrained_effnetb2_feature_extractor_food101_20_percent.pth",
25
  map_location=torch.device("cpu"), # load to CPU
26
  )
27
  )
 
21
  # Load saved weights
22
  effnetb2.load_state_dict(
23
  torch.load(
24
+ f="models/pretrained_effnetb2_feature_extractor_food101_100_percent.pth",
25
  map_location=torch.device("cpu"), # load to CPU
26
  )
27
  )
models/pretrained_effnetb2_feature_extractor_food101_100_percent.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45054b3bf6ef8b14326fa96802659114a3508e87a843c88a470c0a26a0a31a06
3
+ size 31857722