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, 2024

Commit

34d59ab

·

1 Parent(s): 1ea9d7f

added confidence

Files changed (1) hide show

generic_nel.py +1 -1

generic_nel.py CHANGED Viewed

@@ -155,7 +155,7 @@ class NelPipeline(Pipeline):
         # percentages = [(prob * 100.0).tolist() for prob in all_probabilities]
         # Return the predictions along with the extracted entity, lOffset, and rOffset
-        return wikipedia_predictions, enclosed_entity, lOffset, rOffset, percentages
     def _forward(self, inputs):
         return inputs

         # percentages = [(prob * 100.0).tolist() for prob in all_probabilities]
         # Return the predictions along with the extracted entity, lOffset, and rOffset
+        return wikipedia_predictions, enclosed_entity, lOffset, rOffset, [percentages]
     def _forward(self, inputs):
         return inputs