fix arg

app/main.py

@@ -302,6 +302,7 @@ async def handle_cnn(file: UploadFile = File(...)):
 
         # Make predictions
         predictions = cnn_model.predict(X)
+        print('predictions', predictions)
 
         # Convert predictions to label indexes
         predicted_label_indexes = np.argmax(predictions, axis=1)
@@ -310,6 +311,7 @@ async def handle_cnn(file: UploadFile = File(...)):
         predicted_labels = cnn_label_encoder.inverse_transform(predicted_label_indexes)
 
         print('decoded', predicted_labels)
+        print('arg', np.argmax(predicted_labels, axis=1))
         # .tolist()
         # Clean up the temporary file
         os.remove(temp_filename)