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
|