Update app.py

app.py

@@ -12,7 +12,7 @@ class_names = ['pizza','steak','sushi']
 
 # Model and transforms preparation
 # Create EffNetB2 model
-effnetb2 , effnetb2_transforms = create_effnetb2_model(num_classes=len(class_names))
+effnetb2 , effnetb2_transforms = create_effnetb2_model(num_classes=3)
 
 # load and save weights