Update tasks/audio.py

tasks/audio.py

@@ -20,8 +20,8 @@ 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_cloudpickle.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)
@@ -63,10 +63,10 @@ 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)
+    # with open(MODEL_PATH, 'rb') as f:
+    #     clf = pickle.load(f)
     
     predictions = clf.predict(x_test)