embedding mdoel
Browse files
app.py
CHANGED
|
@@ -54,7 +54,8 @@ def bot_message(history, docs):
|
|
| 54 |
|
| 55 |
user_question = history[-1][0]
|
| 56 |
splits = text_splitter.split_documents(docs)
|
| 57 |
-
vectorstore = Chroma.from_documents(documents=splits, embedding=CohereEmbeddings(
|
|
|
|
| 58 |
retriever = vectorstore.as_retriever()
|
| 59 |
|
| 60 |
rag_chain = (
|
|
|
|
| 54 |
|
| 55 |
user_question = history[-1][0]
|
| 56 |
splits = text_splitter.split_documents(docs)
|
| 57 |
+
vectorstore = Chroma.from_documents(documents=splits, embedding=CohereEmbeddings(model="embed-english-light-v3.0",
|
| 58 |
+
cohere_api_key="COHERE_API_KEY"))
|
| 59 |
retriever = vectorstore.as_retriever()
|
| 60 |
|
| 61 |
rag_chain = (
|