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