Spaces:
Running
Running
Update app.py
Browse files
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"
|
|
|
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
|