Pedro Gengo commited on
Commit
4025c5d
·
1 Parent(s): f2b9694

Adding app

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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.inputs.Slider(1, 10, default=5),
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
  )