Kelmoir commited on
Commit
b11d3e4
·
verified ·
1 Parent(s): 831c906

Update app.py

Browse files

corrected module creation call

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ class_names = ['pizza', 'steak', 'sushi']
12
 
13
  ### 2. Model and transforms setup
14
  effnetb2, effnetb2_transforms = create_effnetb2_model(
15
- num_classes=3,
16
  )
17
 
18
  #Load the saved weights
 
12
 
13
  ### 2. Model and transforms setup
14
  effnetb2, effnetb2_transforms = create_effnetb2_model(
15
+ output_classes=3,
16
  )
17
 
18
  #Load the saved weights