kdallash commited on
Commit
f472d67
·
verified ·
1 Parent(s): 1a2d2be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from typing import List, Dict,Tuple
9
  class_names = ["pizza", "steak", "sushi"]
10
 
11
  ### 2 model and transform preparation###
12
- effnetb2_loaded, transforms = create_effnetb2_model(num_classes=len(class_names))
13
  effnetb2_loaded.load_state_dict(torch.load("11-model_deployment_effnetb2.pth", map_location="cpu"))
14
  effnetb2_loaded.to("cpu")
15
 
 
9
  class_names = ["pizza", "steak", "sushi"]
10
 
11
  ### 2 model and transform preparation###
12
+ effnetb2_loaded, effnet_transform = create_effnetb2_model(num_classes=len(class_names))
13
  effnetb2_loaded.load_state_dict(torch.load("11-model_deployment_effnetb2.pth", map_location="cpu"))
14
  effnetb2_loaded.to("cpu")
15