Spaces:
Sleeping
Sleeping
Pedro Gengo
commited on
Commit
·
4025c5d
1
Parent(s):
f2b9694
Adding app
Browse files
app.py
CHANGED
@@ -7,15 +7,18 @@ from sentence_transformers import SentenceTransformer
|
|
7 |
|
8 |
DIM = 768
|
9 |
model = SentenceTransformer("nomic-ai/nomic-embed-text-v1.5", trust_remote_code=True)
|
|
|
10 |
|
11 |
papers_df = pd.read_csv("data/cvpr2024_papers_with_details.csv", index_col=None, on_bad_lines='skip')
|
12 |
papers_df = papers_df[~papers_df["summary"].isna() & ~papers_df["pdf_path"].isna()]
|
|
|
13 |
|
14 |
with open('data/embeddings.npy', 'rb') as f:
|
15 |
embeddings = np.load(f)
|
16 |
|
17 |
index = faiss.IndexFlatL2(DIM)
|
18 |
index.add(embeddings)
|
|
|
19 |
|
20 |
|
21 |
def encode_query(query):
|
@@ -34,7 +37,7 @@ demo = gr.Interface(
|
|
34 |
search_nearest_papers,
|
35 |
[
|
36 |
"text",
|
37 |
-
gr.
|
38 |
],
|
39 |
"dataframe",
|
40 |
)
|
|
|
7 |
|
8 |
DIM = 768
|
9 |
model = SentenceTransformer("nomic-ai/nomic-embed-text-v1.5", trust_remote_code=True)
|
10 |
+
print("Model loaded successfully")
|
11 |
|
12 |
papers_df = pd.read_csv("data/cvpr2024_papers_with_details.csv", index_col=None, on_bad_lines='skip')
|
13 |
papers_df = papers_df[~papers_df["summary"].isna() & ~papers_df["pdf_path"].isna()]
|
14 |
+
print("Data loaded successfully")
|
15 |
|
16 |
with open('data/embeddings.npy', 'rb') as f:
|
17 |
embeddings = np.load(f)
|
18 |
|
19 |
index = faiss.IndexFlatL2(DIM)
|
20 |
index.add(embeddings)
|
21 |
+
print("Index loaded successfully")
|
22 |
|
23 |
|
24 |
def encode_query(query):
|
|
|
37 |
search_nearest_papers,
|
38 |
[
|
39 |
"text",
|
40 |
+
gr.Slider(1, 10, default=5),
|
41 |
],
|
42 |
"dataframe",
|
43 |
)
|