Mehrdad Esmaeili commited on
Commit
acf1331
1 Parent(s): 4818fa5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ texts = text_splitter.split_documents(documents)
41
  embeddings = CohereEmbeddings(model='embed-english-v3.0')
42
  docsearch = Chroma.from_documents(texts, embeddings)
43
 
44
- qa = RetrievalQA.from_chain_type(llm=Cohere(model='base'), chain_type="stuff", \
45
  retriever=docsearch.as_retriever(search_kwargs={'k':1}),return_source_documents=True)
46
 
47
  def predict(message, history):
 
41
  embeddings = CohereEmbeddings(model='embed-english-v3.0')
42
  docsearch = Chroma.from_documents(texts, embeddings)
43
 
44
+ qa = RetrievalQA.from_chain_type(llm=Cohere(model='command'), chain_type="stuff", \
45
  retriever=docsearch.as_retriever(search_kwargs={'k':1}),return_source_documents=True)
46
 
47
  def predict(message, history):