Emanuela Boros commited on
Commit
79e27f6
·
1 Parent(s): 6582c45

remove debug

Browse files
Files changed (1) hide show
  1. generic_nel.py +0 -2
generic_nel.py CHANGED
@@ -114,8 +114,6 @@ class NelPipeline(Pipeline):
114
  return_dict_in_generate=True,
115
  output_scores=True,
116
  )
117
- print(outputs.keys(), "!!!!")
118
- token_ids, scores = outputs.sequences, outputs.scores
119
  # Decode the predictions into readable text
120
  wikipedia_predictions = self.tokenizer.batch_decode(
121
  outputs.sequences, skip_special_tokens=True
 
114
  return_dict_in_generate=True,
115
  output_scores=True,
116
  )
 
 
117
  # Decode the predictions into readable text
118
  wikipedia_predictions = self.tokenizer.batch_decode(
119
  outputs.sequences, skip_special_tokens=True