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()
|