JPLTedCas commited on
Commit
d57a4cc
·
verified ·
1 Parent(s): 41e57b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -127,7 +127,8 @@ def get_vectorstore(text_chunks):
127
  FAISS
128
  A FAISS vector store containing the embeddings of the text chunks.
129
  """
130
- MODEL_NAME = "WhereIsAI/UAE-Large-V1"#"sentence-transformers/all-MiniLM-L6-v2"
 
131
  hf_embeddings = HuggingFaceEmbeddings(model_name=MODEL_NAME)
132
  vectorstore = Chroma.from_documents(text_chunks, hf_embeddings, persist_directory="db")
133
  return vectorstore
 
127
  FAISS
128
  A FAISS vector store containing the embeddings of the text chunks.
129
  """
130
+ MODEL_NAME = "WhereIsAI/UAE-Large-V1"
131
+ MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
132
  hf_embeddings = HuggingFaceEmbeddings(model_name=MODEL_NAME)
133
  vectorstore = Chroma.from_documents(text_chunks, hf_embeddings, persist_directory="db")
134
  return vectorstore