BigTimeCoderSean commited on
Commit
07ae9e6
1 Parent(s): 0c29279

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ with open('class_names.txt', 'r') as f:
16
  #Create effnetb2 model
17
 
18
  #create model and transforms preparation
19
- effnetb2, effnetb2_transforms = create_effnetb2(num_classes = 101)
20
 
21
  effnetb2.load_state_dict(
22
  torch.load(
 
16
  #Create effnetb2 model
17
 
18
  #create model and transforms preparation
19
+ effnetb2, effnetb2_transforms = create_effnet_b2(num_classes = 101)
20
 
21
  effnetb2.load_state_dict(
22
  torch.load(