Update app.py

app.py

@@ -100,8 +100,8 @@ def ant_genus_classification(image):
     print(image.shape)
     with torch.no_grad():
         y_hat = model(image)
-        y_hat = torch.nn.functional.softmax(y_hat, dim=1)
-        preds = torch.argmax(y_hat, dim=1)
+        preds = torch.nn.functional.softmax(y_hat, dim=1)
+        # preds = torch.argmax(preds, dim=1)
         print(preds.shape)
         
         confidences = {class_to_idx[i]: float(preds[i]) for i in range(len(classes))}