Update tasks/text.py

tasks/text.py

@@ -9,6 +9,7 @@ from .utils.emissions import tracker, clean_emissions_data, get_space_info
 
 
 import os
+import logging 
 import numpy as np
 print(os.getcwd())
 #
@@ -19,6 +20,10 @@ import pickle
 import xgboost as xgb
 
 
+#logging 
+logging.basicConfig(level=logging.INFO)
+
+logging.info("LAS ESTRELLAS!!!!!")
 
 router = APIRouter()
 
@@ -67,7 +72,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     #--------------------------------------------------------------------------------------------
     # Load a pre-trained Sentence-BERT model
     print("loading model")
-    model = SentenceTransformer('sentence-transformers/all-mpnet-base-v2', device='cpu')
+    model = SentenceTransformer('sentence-transformers/all-MPNET-base-v2', device='cpu')
     # Generate sentence embeddings
     sentence_embeddings = model.encode(test_dataset["quote"])
 
@@ -88,6 +93,9 @@ async def evaluate_text(request: TextEvaluationRequest):
     #xgb_multi.load_model("xgb_model_muli.bin")
 
 
+
+
+   
     X_train = sentence_embeddings.copy()
     
     y_train = test_dataset["label"].copy()
@@ -101,6 +109,10 @@ async def evaluate_text(request: TextEvaluationRequest):
     X_train_multi = X_train[y_train != 0]
 
     y_train_multi = y_train[y_train != 0]
+    
+    logging.info(f"Xtrain_multi_shape:{X_train_multi.shape}")
+    logging.info(f"Xtrain shape:{X_train.shape}")
+                  
 
     #predictions
     y_pred_bin = xgb_bin.predict(X_train)