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

b828224

•

1 Parent(s): 0dee202

added confidence

Files changed (1) hide show

generic_nel.py +1 -1

generic_nel.py CHANGED Viewed

@@ -127,7 +127,7 @@ class NelPipeline(Pipeline):
         )
         print(len(transition_scores), "!!!!")
         probabilities = transition_scores[0]
-        print(probabilities, "!!!!")
         # percentages = (probabilities * 100.0).cpu().numpy().tolist()

         )
         print(len(transition_scores), "!!!!")
         probabilities = transition_scores[0]
+        print(probabilities, "!!!!", probabilities.shape)
         # percentages = (probabilities * 100.0).cpu().numpy().tolist()