Commit
·
25e0e5e
1
Parent(s):
483a49c
WIP confidence
Browse files- generic_nel.py +2 -0
generic_nel.py
CHANGED
|
@@ -112,6 +112,8 @@ class NelPipeline(Pipeline):
|
|
| 112 |
max_new_tokens=30,
|
| 113 |
)
|
| 114 |
|
|
|
|
|
|
|
| 115 |
# Decode the predictions into readable text
|
| 116 |
wikipedia_predictions = self.tokenizer.batch_decode(
|
| 117 |
outputs, skip_special_tokens=True
|
|
|
|
| 112 |
max_new_tokens=30,
|
| 113 |
)
|
| 114 |
|
| 115 |
+
print(outputs)
|
| 116 |
+
|
| 117 |
# Decode the predictions into readable text
|
| 118 |
wikipedia_predictions = self.tokenizer.batch_decode(
|
| 119 |
outputs, skip_special_tokens=True
|