Spaces:
Runtime error
Runtime error
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
|