fix: correct passage loading and retrieval function call in main.py

main.py

@@ -19,7 +19,7 @@ index_path = hf_hub_download(repo_id='YuITC/Vietnamese-Legal-Doc-Retrieval-Data'
                              local_dir='demo')
 
 emb_model   = SentenceTransformer('YuITC/bert-base-multilingual-cased-finetuned-VNLegalDocs')
-passages    = pd.read_parquet(passages_path)
+passages    = pd.read_parquet(passages_path)['text'].tolist()
 legal_index = faiss.read_index(index_path)
 
 
@@ -41,7 +41,7 @@ def retrieval(emb_model, query, index, top_k=10):
             } for i in range(len(cand_idxs))]
 
 def get_results(query, top_k):
-    hits = retrieval(fine_tuned_model, query, legal_index, top_k=top_k)
+    hits = retrieval(emb_model, query, legal_index, top_k=top_k)
     
     result = ""
     for rank, h in enumerate(hits, start=1):
@@ -68,4 +68,4 @@ demo = gr.Interface(
 )
 
 if __name__ == '__main__':
-    demo.launch(share=True)
+    demo.launch()