Update tasks/text.py

tasks/text.py

@@ -29,8 +29,6 @@ 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()
@@ -98,8 +96,11 @@ 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)
+    try :
+        model_nn = torch.load("model_nn.pth", map_location=device)
+    except :
+        print(os.getcwd())
+        
 
     # Set the model to evaluation mode
     model_nn.eval()