update app.py

app.py

@@ -47,7 +47,7 @@ def pred(img):
     model.eval()
     transformed = manual_transform(img).to(device)
     with torch.inference_mode():
-        logits = model_0(transformed.unsqueeze(dim=0))
+        logits = model(transformed.unsqueeze(dim=0))
         pred = torch.softmax(logits, dim=-1)
     return f"prediction: {class_names[pred.argmax(dim=-1).item()]} | confidence: {pred.max():.3f}"