Update tasks/text.py

tasks/text.py

@@ -16,6 +16,9 @@ from sentence_transformers import SentenceTransformer
 from xgboost import XGBClassifier
 import pickle
 
+import xgboost as xgb
+
+
 
 router = APIRouter()
 
@@ -68,11 +71,15 @@ async def evaluate_text(request: TextEvaluationRequest):
     sentence_embeddings = model.encode(test_dataset["quote"])
 
     #load the models
-    with open("xgb_bin.pkl","rb") as f:
-        xgb_bin = pickle.load(f)
+    
+
+    # Load the binary model
+    xgb_bin = xgb.Booster()
+    xgb_bin.load_model("xgb_model_bin.bin")
 
-    with open("xgb_multi.pkl","rb") as f:
-        xgb_multi = pickle.load(f)
+        # Load the binary model
+    xgb_multi = xgb.Booster()
+    xgb_multi.load_model("xgb_model_multi.bin")
 
 
     X_train = sentence_embeddings.copy()