Emanuela Boros
commited on
Commit
·
79e27f6
1
Parent(s):
6582c45
remove debug
Browse files- 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
|