update indexes

app/main.py

@@ -329,9 +329,9 @@ async def handle_cnn(file: UploadFile = File(...)):
         # Convert predictions to label indexes
         predicted_label_indexes = np.argmax(predictions, axis=1)
         print(predicted_label_indexes)
-        predicted_label_indexes = np.argmax(predicted_label_indexes)
+        # predicted_label_indexes = np.argmax(predicted_label_indexes)
         # 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]