Adding app

app.py

@@ -30,8 +30,8 @@ def encode_query(query):
 
 def search_nearest_papers(query, k=5):
     query_embeddings = encode_query(query)
-    D, I = index.search(query_embeddings.numpy(), k)
-    return papers_df.iloc[I[0]][["Title", "summary", "pdf_path"]]
+    D, I = index.search(query_embeddings, k)
+    return papers_df.iloc[I[0]][["Title", "pdf_path"]]
 
 demo = gr.Interface(
     search_nearest_papers,
@@ -40,7 +40,7 @@ demo = gr.Interface(
         gr.Slider(1, 10, value=5),
     ],
     gr.Dataframe(
-            headers=["Title", "Summary", "PDF"],
+            headers=["Title", "PDF"],
     ),
     title="CVPR 2024 Paper Search",
     description="Semantic search over CPVR 2024 paper summary. This app was made using the data available on https://github.com/harpreetsahota204/CVPR-2024-Papers.",

data/embeddings.npy

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:939ae59573f4807a026ca5eeaa9bafc6daa3cbb6a7fb9b1f163991c6ede3ac09
-size 3200