tarekfer8 commited on
Commit
c396b0b
·
verified ·
1 Parent(s): 029c990

Update djezzy.py

Browse files
Files changed (1) hide show
  1. djezzy.py +5 -1
djezzy.py CHANGED
@@ -14,7 +14,11 @@ from langchain.embeddings import SentenceTransformerEmbeddings
14
  from langchain.vectorstores import FAISS
15
  from sklearn.metrics.pairwise import cosine_similarity
16
  import json
17
- import pickle
 
 
 
 
18
  def load_data(text_filename='docs_text.json', embeddings_filename='docs_embeddings.json'):
19
 
20
 
 
14
  from langchain.vectorstores import FAISS
15
  from sklearn.metrics.pairwise import cosine_similarity
16
  import json
17
+
18
+
19
+ model_name = "sentence-transformers/all-MiniLM-L6-v2"
20
+
21
+ embedding_llm = SentenceTransformerEmbeddings(model_name=model_name)
22
  def load_data(text_filename='docs_text.json', embeddings_filename='docs_embeddings.json'):
23
 
24