Spaces:
Sleeping
Sleeping
Pedro Gengo
commited on
Commit
·
06b44ea
1
Parent(s):
7c0d447
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.numpy(), k)
|
| 34 |
-
return papers_df.iloc[I[0]][["
|
| 35 |
|
| 36 |
demo = gr.Interface(
|
| 37 |
search_nearest_papers,
|
|
@@ -41,7 +41,9 @@ demo = gr.Interface(
|
|
| 41 |
],
|
| 42 |
gr.Dataframe(
|
| 43 |
headers=["Title", "Summary", "PDF"],
|
| 44 |
-
)
|
|
|
|
|
|
|
| 45 |
)
|
| 46 |
|
| 47 |
if __name__ == "__main__":
|
|
|
|
| 31 |
def search_nearest_papers(query, k=5):
|
| 32 |
query_embeddings = encode_query(query)
|
| 33 |
D, I = index.search(query_embeddings.numpy(), k)
|
| 34 |
+
return papers_df.iloc[I[0]][["Title", "summary", "pdf_path"]]
|
| 35 |
|
| 36 |
demo = gr.Interface(
|
| 37 |
search_nearest_papers,
|
|
|
|
| 41 |
],
|
| 42 |
gr.Dataframe(
|
| 43 |
headers=["Title", "Summary", "PDF"],
|
| 44 |
+
),
|
| 45 |
+
title="CVPR 2024 Paper Search",
|
| 46 |
+
description="Semantic search over CPVR 2024 paper summary. This app was made using the data available on https://github.com/harpreetsahota204/CVPR-2024-Papers.",
|
| 47 |
)
|
| 48 |
|
| 49 |
if __name__ == "__main__":
|