ypesk commited on
Commit
830a067
·
verified ·
1 Parent(s): bea7c94

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. 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