Pedro Gengo commited on
Commit
62338e3
1 Parent(s): 62d4f4c

Adding app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def encode_query(query):
31
  def search_nearest_papers(query, k=5):
32
  query_embeddings = encode_query(query)
33
  D, I = index.search(query_embeddings, k)
34
- return papers_df.iloc[I[0]][["Title", "pdf_path"]]
35
 
36
  demo = gr.Interface(
37
  search_nearest_papers,
 
31
  def search_nearest_papers(query, k=5):
32
  query_embeddings = encode_query(query)
33
  D, I = index.search(query_embeddings, k)
34
+ return papers_df.iloc[I[0]][["Title", "arXiv_link"]]
35
 
36
  demo = gr.Interface(
37
  search_nearest_papers,