Himel09 commited on
Commit
73fc7de
·
verified ·
1 Parent(s): 51124c8

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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
- embeddings = 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
 
 
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