Update tasks/text.py

tasks/text.py

@@ -66,7 +66,8 @@ async def evaluate_text(request: TextEvaluationRequest):
 
     #--------------------------------------------------------------------------------------------
     # Load a pre-trained Sentence-BERT model
-    model = SentenceTransformer('sentence-transformers/all-MPNET-base-v2', device='cpu')
+    print("loading model")
+    model = SentenceTransformer('sentence-transformers/all-mpnet-base-v2', device='cpu')
     # Generate sentence embeddings
     sentence_embeddings = model.encode(test_dataset["quote"])
 
@@ -79,7 +80,7 @@ async def evaluate_text(request: TextEvaluationRequest):
 
         # Load the binary model
     xgb_multi = xgb.Booster()
-    xgb_multi.load_model("xgb_model_multi.bin")
+    xgb_multi.load_model("xgb_model_muli.bin")
 
 
     X_train = sentence_embeddings.copy()