BrandonFors commited on
Commit
9d79386
·
1 Parent(s): 742adc7

fixed model state path

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(num_classes=3)
16
 
17
  # Load saved weights
18
 
19
- effnetb2.load_state_dict(torch.load(f="foodvision_mini/pretrained_effnetb2_feature_extractor_pizza_steak_sushi_20_percent.pth",
20
  map_location=torch.device("cpu")))
21
 
22
  ### 3. Predict function ###
@@ -60,4 +60,4 @@ demo = gr.Interface(fn=predict, # maps inputs to outputs
60
  description=description,
61
  article=article)
62
  # launch the demo
63
- demo.launch()
 
16
 
17
  # Load saved weights
18
 
19
+ effnetb2.load_state_dict(torch.load(f="pretrained_effnetb2_feature_extractor_pizza_steak_sushi_20_percent.pth",
20
  map_location=torch.device("cpu")))
21
 
22
  ### 3. Predict function ###
 
60
  description=description,
61
  article=article)
62
  # launch the demo
63
+ demo.launch(debug=False)