sabazo commited on
Commit
e9df42e
·
1 Parent(s): b11f76e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ s3.download_file('rad-rag-demos', 'vectorstores/chroma.sqlite3', './chroma_db/ch
25
 
26
  db = Chroma(persist_directory="./chroma_db", embedding_function=embeddings)
27
  db.get()
28
- retriever = db.as_retriever(search_type = "mmr")
29
 
30
  global qa
31
  qa = RetrievalQA.from_chain_type(llm=model_id, chain_type="stuff", retriever=retriever)
 
25
 
26
  db = Chroma(persist_directory="./chroma_db", embedding_function=embeddings)
27
  db.get()
28
+ retriever = db.as_retriever()
29
 
30
  global qa
31
  qa = RetrievalQA.from_chain_type(llm=model_id, chain_type="stuff", retriever=retriever)