fix issue 3

09_pretrained_effnetb2_feature_extractor_pizza_steak_sushi_20_percent.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:02b1a2d0a83e37ff082345bdb3ee37bc0a9d90d594fb71128a8d2ba18f943754
-size 31307450
+oid sha256:49f172e8691ca003797f29f904dccfee4dd0d1aa99382313c75915a1fffa7a3b
+size 31273033

model.py

@@ -1,7 +1,7 @@
 import torchvision
 import torch
 import torch.nn as nn
-def create_effnetb2_model(num_classes=3, seed=42):
+def create_effnetb2_model(num_classes: int=3, seed: int=42):
     # 1, 2, 3. Create EffNetB2 pretrained weights, transforms and model
     weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT
     transforms = weights.transforms()