NaolTaye commited on
Commit
0e65d94
·
1 Parent(s): 223c003
Files changed (1) hide show
  1. tasks/text.py +1 -1
tasks/text.py CHANGED
@@ -82,7 +82,7 @@ async def evaluate_text(request: TextEvaluationRequest):
82
  # dataloader = DataLoader(tokenized_test, batch_size=16, shuffle=False, collate_fn=data_collator)
83
 
84
  print("Started prediction run")
85
- tokenized_test = tokenizer(test_dataset['quote'], padding=True, truncation=True, return_tensors='pt')
86
  # Model inference
87
  model.eval()
88
  predictions = np.array([])
 
82
  # dataloader = DataLoader(tokenized_test, batch_size=16, shuffle=False, collate_fn=data_collator)
83
 
84
  print("Started prediction run")
85
+ # tokenized_test = tokenizer(test_dataset['quote'], padding=True, truncation=True, return_tensors='pt')
86
  # Model inference
87
  model.eval()
88
  predictions = np.array([])