Update tasks/text.py

tasks/text.py

@@ -102,7 +102,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     current_dir = os.path.dirname(current_file_path)
 
     # model_nn = torch.load(os.path.join(current_dir,"model_nn.pth"), map_location=device)
-    model_nn = torch.load(os.path.join(current_dir,"model_nn_scripted.pth"), map_location=device)  
+    model_nn = torch.jit.load(os.path.join(current_dir,"model_nn_scripted.pth"), map_location=device)  
         
 
     # Set the model to evaluation mode