impresso-project
/

nel-mgenre-multilingual

Text2Text Generation

entity-retrieval

named-entity-disambiguation

entity-disambiguation

named-entity-linking

Inference Endpoints

Model card Files Files and versions Community

Emanuela Boros commited on Oct 18

Commit

8cda137

•

1 Parent(s): 415d945

added confidence

Files changed (1) hide show

generic_nel.py +1 -1

generic_nel.py CHANGED Viewed

@@ -118,7 +118,7 @@ class NelPipeline(Pipeline):
         token_ids, scores = outputs.sequences, outputs.scores
         # Decode the predictions into readable text
         wikipedia_predictions = self.tokenizer.batch_decode(
-            outputs, skip_special_tokens=True
         )
         print(len(outputs.sequences))
         print(outputs.sequences.shape)

         token_ids, scores = outputs.sequences, outputs.scores
         # Decode the predictions into readable text
         wikipedia_predictions = self.tokenizer.batch_decode(
+            outputs.sequences, skip_special_tokens=True
         )
         print(len(outputs.sequences))
         print(outputs.sequences.shape)