valeriedaash commited on
Commit
d717b7c
·
1 Parent(s): 5ffc08d

vectors added

Browse files
Files changed (3) hide show
  1. .gitattributes +1 -0
  2. app.py +2 -2
  3. vectors.txt +3 -0
.gitattributes CHANGED
@@ -36,3 +36,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
36
  embeddings.txt filter=lfs diff=lfs merge=lfs -text
37
  data_final.csv filter=lfs diff=lfs merge=lfs -text
38
  data_prev.csv filter=lfs diff=lfs merge=lfs -text
 
 
36
  embeddings.txt filter=lfs diff=lfs merge=lfs -text
37
  data_final.csv filter=lfs diff=lfs merge=lfs -text
38
  data_prev.csv filter=lfs diff=lfs merge=lfs -text
39
+ vectors.txt filter=lfs diff=lfs merge=lfs -text
app.py CHANGED
@@ -21,7 +21,7 @@ def embed_bert_cls(text, model, tokenizer):
21
  embeddings = torch.nn.functional.normalize(embeddings)
22
  return embeddings[0].cpu().numpy()
23
 
24
- books_embs = np.loadtxt('embeddings.txt')
25
 
26
  index = faiss.IndexFlatIP(books_embs.shape[1])
27
  index.add(books_embs)
@@ -46,7 +46,7 @@ if text and recommend_button:
46
  st.image(df['image_url'][i], use_column_width=True)
47
  st.write(round(j* 100, 2))
48
  with col_2:
49
- st.write(f'Название книги: {df["title"][i]}')
50
  st.write(f'Автор: {df["author"][i]}')
51
  st.write(f'Ссылка: {df["page_url"][i]}')
52
  st.write(f'Аннотация: {df["annotation"][i]}')
 
21
  embeddings = torch.nn.functional.normalize(embeddings)
22
  return embeddings[0].cpu().numpy()
23
 
24
+ books_embs = np.loadtxt('vectors.txt')
25
 
26
  index = faiss.IndexFlatIP(books_embs.shape[1])
27
  index.add(books_embs)
 
46
  st.image(df['image_url'][i], use_column_width=True)
47
  st.write(round(j* 100, 2))
48
  with col_2:
49
+ st.write(f'Название книги: **{df["title"][i]}**')
50
  st.write(f'Автор: {df["author"][i]}')
51
  st.write(f'Ссылка: {df["page_url"][i]}')
52
  st.write(f'Аннотация: {df["annotation"][i]}')
vectors.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b98252d0ec0d8e730da90d18b70da597f6ab2bac583f277e3f44437d90e69c34
3
+ size 141223994