aur-beau commited on
Commit
089ac66
·
1 Parent(s): 87bc31c

fixed path

Browse files
Files changed (1) hide show
  1. tasks/text.py +1 -1
tasks/text.py CHANGED
@@ -95,7 +95,7 @@ async def evaluate_text(request: TextEvaluationRequest):
95
  x = self.fc3(x) # Output raw logits
96
  return x
97
 
98
- model_nn = torch.load("../tasks/models/model_nn.pth", map_location=device)
99
 
100
  # Set the model to evaluation mode
101
  model_nn.eval()
 
95
  x = self.fc3(x) # Output raw logits
96
  return x
97
 
98
+ model_nn = torch.load("models/model_nn.pth", map_location=device)
99
 
100
  # Set the model to evaluation mode
101
  model_nn.eval()