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,
|