Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
|
@@ -161,7 +161,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
| 161 |
|
| 162 |
with torch.no_grad():
|
| 163 |
if MODEL =="mlp":
|
| 164 |
-
b_texts = batch
|
| 165 |
logits = model(b_texts)
|
| 166 |
elif MODEL == "ct":
|
| 167 |
b_input_ids, b_input_mask, b_token_type_ids, b_labels = batch
|
|
|
|
| 161 |
|
| 162 |
with torch.no_grad():
|
| 163 |
if MODEL =="mlp":
|
| 164 |
+
b_texts = batch[0]
|
| 165 |
logits = model(b_texts)
|
| 166 |
elif MODEL == "ct":
|
| 167 |
b_input_ids, b_input_mask, b_token_type_ids, b_labels = batch
|