Update tasks/text.py

tasks/text.py

@@ -12,7 +12,7 @@ from sentence_transformers import SentenceTransformer
 from xgboost import XGBClassifier
 import pickle
 import torch
-
+import os
 
 router = APIRouter()
 
@@ -29,6 +29,9 @@ async def evaluate_text(request: TextEvaluationRequest):
     - Makes random predictions from the label space (0-7)
     - Used as a baseline for comparison
     """
+
+    print(os.getcwd())
+    
     # Get space info
     username, space_url = get_space_info()
 
@@ -95,6 +98,7 @@ async def evaluate_text(request: TextEvaluationRequest):
             x = self.fc3(x)  # Output raw logits
             return x
 
+    
     model_nn = torch.load("model_nn.pth", map_location=device)
 
     # Set the model to evaluation mode