last time

app.py

@@ -21,9 +21,9 @@ def predict(img):
     
     img = transform(img).unsqueeze(0)
 
-    model.eval()
+    effnetb2.eval()
     with torch.inference_mode():
-        pred_label = class_names[model(img).softmax(dim = 1).argmax(dim = 1)]
+        pred_label = class_names[effnetb2(img).softmax(dim = 1).argmax(dim = 1)]
     print(pred_label)
     return pred_label