Update tasks/audio.py

tasks/audio.py

@@ -5,6 +5,7 @@ from sklearn.metrics import accuracy_score
 import os
 import joblib
 from pathlib import Path
+import pickle
 
 from sklearn.preprocessing import StandardScaler
 from .utils.fourier import FourierPreprocessor, Fourier
@@ -19,7 +20,9 @@ router = APIRouter()
 
 DESCRIPTION = "Random Forest"
 ROUTE = "/audio"
-MODEL_PATH = Path(__file__).parent / "audio_models" / "RandomForestClassifier_withScaler.pkl"
+# MODEL_PATH = Path(__file__).parent / "audio_models" / "RandomForestClassifier_withScaler.pkl"
+MODEL_PATH = Path(__file__).parent / "audio_models" / "RandomForestClassifier_withScaler_cloudpickle.pkl"
+
 
 @router.post(ROUTE, tags=["Audio Task"], description=DESCRIPTION)
 async def evaluate_audio(request: AudioEvaluationRequest):
@@ -60,7 +63,11 @@ async def evaluate_audio(request: AudioEvaluationRequest):
     # Extract audio samples from test_dataset
     x_test = [sample["audio"]["array"] for sample in test_dataset]
 
-    clf = joblib.load(MODEL_PATH)
+    # clf = joblib.load(MODEL_PATH)
+
+    with open(MODEL_PATH, 'rb') as f:
+        clf = pickle.load(f)
+    
     predictions = clf.predict(x_test)
 
     # --------------------------------------------------------------------------------------------