fix argmax

app/main.py

@@ -327,8 +327,9 @@ async def handle_cnn(file: UploadFile = File(...)):
         print('predictions', predictions)
 
         # Convert predictions to label indexes
-        predicted_label_indexes = np.argmax(predictions)
-
+        predicted_label_indexes = np.argmax(predictions, axis=1)
+        print(predicted_label_indexes)
+        
         # Convert label indexes to actual label names
         predicted_labels = cnn_label_encoder.inverse_transform([predicted_label_indexes])