Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
|
@@ -93,7 +93,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
| 93 |
outputs = model(input_ids, attention_mask=attention_mask)
|
| 94 |
preds = torch.argmax(outputs.logits, dim=1)
|
| 95 |
predictions.extend(preds.cpu().numpy())
|
| 96 |
-
print('here
|
| 97 |
|
| 98 |
true_labels = test_dataset["label"]
|
| 99 |
|
|
|
|
| 93 |
outputs = model(input_ids, attention_mask=attention_mask)
|
| 94 |
preds = torch.argmax(outputs.logits, dim=1)
|
| 95 |
predictions.extend(preds.cpu().numpy())
|
| 96 |
+
print('here is a batch')
|
| 97 |
|
| 98 |
true_labels = test_dataset["label"]
|
| 99 |
|