Update ✨Entity Linking Application✨.py

✨Entity Linking Application✨.py

@@ -436,8 +436,8 @@ 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:
+                            my_bar = st.progress(0)
+                            for i, element in enumerate(json_file):
                                 qid = element.get("qid")
                                 link = f"https://www.wikidata.org/wiki/{qid}"
                                 label = element.get("label")
@@ -447,7 +447,7 @@ def main_cli():
                                 desc_emb = model.encode([description])
                                 
                                 lista.append({link: [label_emb, desc_emb]})
-                                my_bar.progress(len(element) + 1)
+                                my_bar.progress((i + 1) / len(json_file))
                                 print(qid)
     
                             label_dataset_emb = model.encode([i])