Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
|
@@ -171,7 +171,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
| 171 |
b_texts = batch[0]
|
| 172 |
logits = model(b_texts)
|
| 173 |
elif MODEL == "ct":
|
| 174 |
-
b_input_ids, b_input_mask, b_token_type_ids
|
| 175 |
logits = model(b_input_ids, b_token_type_ids, b_input_mask)
|
| 176 |
|
| 177 |
logits = logits.detach().cpu().numpy()
|
|
|
|
| 171 |
b_texts = batch[0]
|
| 172 |
logits = model(b_texts)
|
| 173 |
elif MODEL == "ct":
|
| 174 |
+
b_input_ids, b_input_mask, b_token_type_ids = batch
|
| 175 |
logits = model(b_input_ids, b_token_type_ids, b_input_mask)
|
| 176 |
|
| 177 |
logits = logits.detach().cpu().numpy()
|