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

emanuelaboros commited on 6 days ago

Commit

6111322

·

verified ·

1 Parent(s): 5bcae76

Update generic_nel.py

Files changed (1) hide show

generic_nel.py +1 -1

generic_nel.py CHANGED Viewed

@@ -151,7 +151,7 @@ class NelPipeline(Pipeline):
         sequence_confidence = torch.exp(log_prob_sum)
         percentage = sequence_confidence.cpu().numpy() * 100.0
-        print(wikipedia_prediction, enclosed_entity, lOffset, rOffset, percentage)
         # Return the predictions along with the extracted entity, lOffset, and rOffset
         return wikipedia_prediction, enclosed_entity, lOffset, rOffset, percentage

         sequence_confidence = torch.exp(log_prob_sum)
         percentage = sequence_confidence.cpu().numpy() * 100.0
+        # print(wikipedia_prediction, enclosed_entity, lOffset, rOffset, percentage)
         # Return the predictions along with the extracted entity, lOffset, and rOffset
         return wikipedia_prediction, enclosed_entity, lOffset, rOffset, percentage