try to fix output

app/main.py

@@ -269,15 +269,16 @@ async def handle_audio(file: UploadFile = File(...)):
 
         # proceed with an inference
         results = mlp_model.predict(features)
+        decoded_predictions = [mlp_label_encoder.classes_[i] for i in results]
 
-        # Decode the predictions using the label encoder
-        decoded_predictions = mlp_label_encoder.inverse_transform(results)
-
+        # # Decode the predictions using the label encoder
+        # decoded_predictions = mlp_label_encoder.inverse_transform(results)
+        # .tolist()
         # Clean up the temporary file
         os.remove(temp_filename)
 
         # Return a successful response with decoded predictions
-        return {"message": "File processed successfully", "prediction": decoded_predictions.tolist()}
+        return {"message": "File processed successfully", "prediction": decoded_predictions}
     except Exception as e:
         # Handle possible exceptions
         raise HTTPException(status_code=500, detail=str(e))