Spaces:
Sleeping
Sleeping
fixed path
Browse files- 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("
|
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()
|