Update ✨Entity Linking Application✨.py

✨Entity Linking Application✨.py

@@ -286,7 +286,7 @@ def main_cli():
     st.title("✨ Entity Linking Application ✨")
     st.caption("This web application is part of my master’s dissertation.")
 
-    if "Run Entity Linking" not in st.session_state:
+    if "disabled" not in st.session_state:
         st.session_state.disabled = False
 
     def disable_inputs():
@@ -436,6 +436,7 @@ def main_cli():
                             json_file = json.load(f)
                             lista = []
                             lista_1 = []
+                            my_bar = st.progress(0, text={len(json_file)})
                             for element in json_file:
                                 qid = element.get("qid")
                                 link = f"https://www.wikidata.org/wiki/{qid}"
@@ -446,6 +447,7 @@ def main_cli():
                                 desc_emb = model.encode([description])
                                 
                                 lista.append({link: [label_emb, desc_emb]})
+                                my_bar.progress(len(element) + 1)
                                 print(qid)
     
                             label_dataset_emb = model.encode([i])