second commit

app.py

@@ -11,7 +11,7 @@ from typing import Dict, Tuple
 class_names = ['pizza', 'steak', 'sushi']
 
 ### 2. Model and transforms perparation ###
-effnetb2, effnetb2_transforms = create_effnetb2_model(num_classes=len(class_names))
+effnetb2, effnetb2_transforms = create_effnetb2_model()
 
 # Load save weights
 effnetb2.load_state_dict(torch.load(f='09_pretrained_effnetb2_feature_extractor_pizza_steak_sushi_20_percent.pth',