added .gitignore

.gitignore

@@ -0,0 +1 @@
+__pycache__

09_pretrained_effnetb2_20_percent.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:82ad594c0f96d959e61d7828cf44760f9139b0e4f7c485764c1f0a368cd9241f
-size 31280955

app.py

@@ -11,8 +11,8 @@ class_names = ["pizza", "steak", "sushi"]
 
 effnetb2, effnetb2_transforms = create_effnet2_model(num_classes=3)
 sd = load_file("09_pretrained_effnetb2_20_percent.safetensors")
-missing, unexpected = effnetb2.load_state_dict(sd, strict=True)
-
+effnetb2.load_state_dict(sd, strict=True)
+effnetb2.eval()
 
 
 def predict(img) -> Tuple[Dict, float]:

model.py

@@ -5,7 +5,7 @@ from torch import nn
 def create_effnet2_model(num_classes: int=3):
     weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT
     transforms = weights.transforms()
-    model = torchvision.models.efficientnet_b2(weights=weights)
+    model = torchvision.models.efficientnet_b2(weights=None)
 
     for param in model.parameters():
         param.requires_grad = False