Update main.py

main.py

@@ -12,7 +12,7 @@ model = sentence_transformers.SentenceTransformer('allenai-specter')
 
 full_data = datasets.load_dataset("ccm/publications")['train'].to_pandas()
 
-filter = ["\"abstract\": null" in json.dumps(bibdict) for bibdict in full_data['bibtex'].values]
+filter = ["\"abstract\": null" in json.dumps(bibdict) for bibdict in full_data['bib_dict'].values]
 data = full_data[~pandas.Series(filter)]
 data.reset_index(inplace=True)
 
@@ -30,8 +30,8 @@ def search(query, k):
     search_results = ""
 
     for i in range(k):
-        search_results += '### ' + top_five["bibtex"].values[i]["title"] + '\n'
-        search_results += top_five["bibtex"].values[i]["citation"]
+        search_results += '### ' + top_five["bib_dict"].values[i]["title"] + '\n'
+        search_results += top_five["bib_dict"].values[i]["citation"]
         if top_five["pub_url"].values[i] is not None:
             search_results += ", [Paper](" + top_five["pub_url"].values[i] + ")"
         search_results += "\t\n```\n"