Update ✨Entity Linking Application✨.py

✨Entity Linking Application✨.py

@@ -63,7 +63,7 @@ async def combination_method(name, session):
         for i in x:
             new_word = (i[0] + " " + i[1])
             url = f"{new_word} site:en.wikipedia.org inurl:/wiki/ -inurl:? -inurl:Category: -inurl:Help: -inurl:Special: -inurl:File:"
-            s = search(url, num_results = 12, lang="en", region="us", sleep_interval=5)
+            s = search(url, num_results = 12, lang="en", region="us", sleep_interval=2)
             for i in s:
                 data.add(i.split("/")[-1])
     return data
@@ -74,7 +74,7 @@ async def single_method(name, session):
         new_name = name.replace("-", " ").replace("/", " ").split()
         for i in new_name:
             url = f"{i} site:en.wikipedia.org inurl:/wiki/ -inurl:? -inurl:Category: -inurl:Help: -inurl:Special: -inurl:File:"
-            s = search(url, num_results = 12, lang="en", region="us", sleep_interval=5)
+            s = search(url, num_results = 12, lang="en", region="us", sleep_interval=2)
             for i in s:
                 data.add(i.split("/")[-1])
     return data
@@ -86,7 +86,7 @@ async def mains(name, single, combi):
     
     async with aiohttp.ClientSession() as session:
         url = f"{name} site:en.wikipedia.org inurl:/wiki/ -inurl:? -inurl:Category: -inurl:Help: -inurl:Special: -inurl:File:"
-        s = search(url, num_results = 30, lang="en", region="us", sleep_interval=5)
+        s = search(url, num_results = 30, lang="en", region="us", sleep_interval=2)
         for i in s:
             data.add(i.split("/")[-1])
 
@@ -422,7 +422,7 @@ def main_cli():
                 number = 0
                 for i,j,o in zip(list_with_full_names,list_with_contexts,list_with_names_to_show):
                     number += 1
-                    with st.spinner(f"Applying Candidate Selection module... (4/5) [{number}/{len(list_with_full_names)}]"):
+                    with st.spinner(f"Applying Candidate Selection module... (4/5) [{number}/{len(list_with_full_names)}] (This may take a while)"):
                         with open(f"/home/user/app/info_extraction/{i}.json", "r") as f:
                             json_file = json.load(f)
                             lista = []
@@ -448,7 +448,6 @@ def main_cli():
                                     desc_label = model.similarity(desc_dataset_emb, v[1][0])
                                     emb_mean = np.mean([cossim_label, desc_label])
                                     lista_1.append({k: emb_mean})
-                                    print(k)
     
                             sorted_data = sorted(lista_1, key=lambda x: list(x.values())[0], reverse=True)
                         st.write(f"✅ Applined Candidate Selection module (4/5) [{number}/{len(list_with_full_names)}]")