inverse

app/main.py

@@ -330,7 +330,7 @@ async def handle_cnn(file: UploadFile = File(...)):
         predicted_label_indexes = np.argmax(predictions)
 
         # Convert label indexes to actual label names
-        predicted_labels = cnn_label_encoder.inverse_transform([predicted_label_indexes])
+        predicted_labels = cnn_label_encoder.inverse_transform(predicted_label_indexes)
 
         print('decoded', predicted_labels)
         reciter_name = predicted_labels[0]