added lang

generic_nel.py

@@ -52,11 +52,11 @@ def get_wikipedia_page_props(input_str: str):
                 if "wikibase_item" in page_props:
                     return page_props["wikibase_item"]
                 else:
-                    return qid
+                    return qid, language
             else:
-                return qid
+                return qid, language
     except Exception as e:
-        return qid
+        return qid, language
 
 
 def get_wikipedia_title(qid, language="en"):
@@ -159,11 +159,11 @@ class NelPipeline(Pipeline):
         results = []
         for idx, wikipedia_name in enumerate(wikipedia_predictions):
             # Get QID
-            qid = get_wikipedia_page_props(wikipedia_name)
+            qid, language = get_wikipedia_page_props(wikipedia_name)
             # print(f"{wikipedia_name} -- QID: {qid}")
 
             # Get Wikipedia title and URL
-            wkpedia_pagename, url = get_wikipedia_title(qid)
+            wkpedia_pagename, url = get_wikipedia_title(qid, language)
             results.append(
                 {
                     # "id": f"{lOffset}:{rOffset}:{enclosed_entity}:{NEL_MODEL}",