Waflon commited on
Commit
f6cd8e7
1 Parent(s): 8400323

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -25,7 +25,8 @@ def split_text_into_chunks(documents):
25
 
26
  # Function to create embeddings
27
  def create_embeddings():
28
- embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", model_kwargs={'device': "cpu"})
 
29
  return embeddings
30
 
31
  # Function to create vector store
 
25
 
26
  # Function to create embeddings
27
  def create_embeddings():
28
+ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/multi-qa-MiniLM-L6-cos-v1, model_kwargs={'device': "cpu"})
29
+ #embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", model_kwargs={'device': "cpu"})
30
  return embeddings
31
 
32
  # Function to create vector store