Spaces:
Sleeping
Sleeping
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 |
|