Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -1
src/streamlit_app.py
CHANGED
|
@@ -63,7 +63,7 @@ if "processed" not in st.session_state:
|
|
| 63 |
texts = splitter.split_documents(documents)
|
| 64 |
|
| 65 |
#embedding = OllamaEmbeddings(model="mxbai-embed-large")
|
| 66 |
-
|
| 67 |
vectorstore = Chroma.from_documents(documents=texts, embedding=embedding)
|
| 68 |
retriever = vectorstore.as_retriever(search_type="similarity", search_kwargs={"k": 4})
|
| 69 |
|
|
|
|
| 63 |
texts = splitter.split_documents(documents)
|
| 64 |
|
| 65 |
#embedding = OllamaEmbeddings(model="mxbai-embed-large")
|
| 66 |
+
embedding = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
|
| 67 |
vectorstore = Chroma.from_documents(documents=texts, embedding=embedding)
|
| 68 |
retriever = vectorstore.as_retriever(search_type="similarity", search_kwargs={"k": 4})
|
| 69 |
|