Update tasks/text.py

tasks/text.py

@@ -72,15 +72,20 @@ 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)
+        
+    with open("xgb_multi.pkl","rb") as f:
+        xgb_multi = pickle.load(f)
     
 
     # Load the binary model
-    xgb_bin = xgb.Booster()
-    xgb_bin.load_model("xgb_model_bin.bin")
+    #xgb_bin = xgb.Booster()
+    #xgb_bin.load_model("xgb_model_bin.bin")
 
         # Load the binary model
-    xgb_multi = xgb.Booster()
-    xgb_multi.load_model("xgb_model_muli.bin")
+    #xgb_multi = xgb.Booster()
+    #xgb_multi.load_model("xgb_model_muli.bin")
 
 
     X_train = sentence_embeddings.copy()