Update app.py

app.py

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