Spaces:
Sleeping
Sleeping
Pedro Gengo
commited on
Commit
•
62338e3
1
Parent(s):
62d4f4c
Adding app
Browse files
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", "
|
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,
|