Update main.py

main.py

@@ -9,8 +9,8 @@ model = sentence_transformers.SentenceTransformer('allenai-specter')
 
 full_data = datasets.load_dataset("ccm/publications")['train'].to_pandas()
 
-substring = "Designing a New AI"
-filter = full_data['bibtex'].str.contains(substring)
+substring = "0P9w_S0AAAAJ:yB1At4FlUx8C"
+filter = full_data['author_pub_id'].str.contains(substring)
 data = full_data[~filter]
 
 dimensionality = len(data['embedding'][0])
@@ -30,7 +30,7 @@ def search(query):
     for i in range(k):
         search_results += str(i+1) + ". "
         search_results += '"' + top_five["bibtex"].values[i]["title"] + '" '
-        search_results += top_five["bibtex"].values[i]["citation"]
+        search_results += "_" + top_five["bibtex"].values[i]["citation"] + "_"
         if top_five["pub_url"].values[i] is not None:
             search_results += " [Full Paper](" + top_five["pub_url"].values[i] + ")"
         search_results += "\n"