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 19

Commit

71e1eca

•

1 Parent(s): 79e27f6

url --> wk_url

Files changed (2) hide show

.DS_Store +0 -0
generic_nel.py +2 -2

.DS_Store ADDED Viewed

Binary file (6.15 kB). View file

generic_nel.py CHANGED Viewed

@@ -168,9 +168,9 @@ class NelPipeline(Pipeline):
                 {
                     # "id": f"{lOffset}:{rOffset}:{enclosed_entity}:{NEL_MODEL}",
                     "surface": enclosed_entity,
-                    "wkpedia_pagename": wkpedia_pagename,
                     "wkd_id": qid,
-                    "url": url,
                     "type": "UNK",
                     "confidence_nel": round(percentages[idx], 2),
                     "lOffset": lOffset,

                 {
                     # "id": f"{lOffset}:{rOffset}:{enclosed_entity}:{NEL_MODEL}",
                     "surface": enclosed_entity,
                     "wkd_id": qid,
+                    "wkpedia_pagename": wkpedia_pagename,
+                    "wkpedia_url": url,
                     "type": "UNK",
                     "confidence_nel": round(percentages[idx], 2),
                     "lOffset": lOffset,