svsaurav95 commited on
Commit
941089e
·
verified ·
1 Parent(s): c725c88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -74,7 +74,9 @@ def extract_and_embed_text(pdf_file):
74
  docs.append(chunk)
75
  tokenized_texts.append(chunk.split())
76
 
77
- embeddings = embedding_model.encode(docs, batch_size=64, convert_to_numpy=True, normalize_embeddings=True)
 
 
78
 
79
  embedding_dim = embeddings.shape[1]
80
  index = faiss.IndexHNSWFlat(embedding_dim, 32)
 
74
  docs.append(chunk)
75
  tokenized_texts.append(chunk.split())
76
 
77
+ embedding_model = SentenceTransformer("baconnier/Finance2_embedding_small_en-V1.5")
78
+ embedding_model.to(torch.device(device))
79
+
80
 
81
  embedding_dim = embeddings.shape[1]
82
  index = faiss.IndexHNSWFlat(embedding_dim, 32)