Clement Vachet commited on
Commit
c8e3c55
·
1 Parent(s): 1daf9a1

Update HF embedding for multiple languages

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -59,9 +59,9 @@ def create_db(splits, collection_name):
59
  """Create embeddings and vector database"""
60
 
61
  embedding = HuggingFaceEmbeddings(
62
- # model_name="sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
63
  # model_name="sentence-transformers/all-MiniLM-L6-v2",
64
- # model_kwargs={'device': 'cpu'},
65
  # encode_kwargs={'normalize_embeddings': False}
66
  )
67
  new_client = chromadb.EphemeralClient()
 
59
  """Create embeddings and vector database"""
60
 
61
  embedding = HuggingFaceEmbeddings(
62
+ model_name="sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
63
  # model_name="sentence-transformers/all-MiniLM-L6-v2",
64
+ model_kwargs={'device': 'cpu'},
65
  # encode_kwargs={'normalize_embeddings': False}
66
  )
67
  new_client = chromadb.EphemeralClient()