Update app.py
Browse files
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':
|
| 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 |
+
|