focusit commited on
Commit
f9d56ae
·
1 Parent(s): e11d270

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -21,11 +21,13 @@ embeddings = HuggingFaceEmbeddings()
21
  doc_search = Chroma.from_documents(docs, embeddings)
22
 
23
  repo_id = "tiiuae/falcon-7b"
24
- llm = HuggingFaceHub(repo_id=repo_id, huggingfacehub_api_token=hub_token, model_kwargs={'temperature': 0.2,'min_length': 6000})
25
 
26
  from langchain.schema import retriever
27
  retireval_chain = RetrievalQA.from_chain_type(llm, chain_type="stuff", retriever=doc_search.as_retriever())
28
 
29
  if query := st.chat_input("Enter your query "):
30
  with st.chat_message("Assistant"):
31
- st.write(retireval_chain.run(query))
 
 
 
21
  doc_search = Chroma.from_documents(docs, embeddings)
22
 
23
  repo_id = "tiiuae/falcon-7b"
24
+ llm = HuggingFaceHub(repo_id=repo_id, huggingfacehub_api_token=hub_token, model_kwargs={'temperature': 0.2,'min_length': 4000})
25
 
26
  from langchain.schema import retriever
27
  retireval_chain = RetrievalQA.from_chain_type(llm, chain_type="stuff", retriever=doc_search.as_retriever())
28
 
29
  if query := st.chat_input("Enter your query "):
30
  with st.chat_message("Assistant"):
31
+ st.write(retireval_chain.run(query))
32
+
33
+