fix: variable

app.py

@@ -38,7 +38,7 @@ def predict(img) -> Tuple[Dict, float]:
     img = densenet121_transforms(img).unsqueeze(0)
 
     # Put model into evaluation mode and turn on inference mode
-    effnetb2.eval()
+    densenet121.eval()
     with torch.inference_mode():
         # Pass the transformed image through the model and turn the prediction logits into prediction probabilities
         pred_probs = torch.softmax(effnetb2(img), dim=1)