Update app.py

app.py

@@ -9,7 +9,7 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
 text_splitter = RecursiveCharacterTextSplitter(chunk_size=350, chunk_overlap=10)
 
 from langchain.llms import HuggingFaceHub
-model_id = HuggingFaceHub(repo_id="HuggingFaceH4/zephyr-7b-beta", model_kwargs={"temperature":0.1, "max_new_tokens":300})
+model_id = HuggingFaceHub(repo_id="HuggingFaceH4/zephyr-7b-beta", model_kwargs={"temperature":0.1, "max_new_tokens":1024})
 
 from langchain.embeddings import HuggingFaceHubEmbeddings
 embeddings = HuggingFaceHubEmbeddings()
@@ -17,6 +17,7 @@ embeddings = HuggingFaceHubEmbeddings()
 from langchain.vectorstores import Chroma
 
 from langchain.chains import RetrievalQA
+from langchain.chains import RetrievalQAWithSourcesChain
 
 from langchain.prompts import ChatPromptTemplate
 
@@ -34,7 +35,7 @@ db.get()
 retriever = db.as_retriever()
 
 global qa 
-qa = RetrievalQA.from_chain_type(llm=model_id, chain_type="stuff", retriever=retriever, return_source_documents=True)
+qa = RetrievalQAWithSourcesChain.from_chain_type(llm=model_id, chain_type="stuff", retriever=retriever, return_source_documents=True)
 
 
 def add_text(history, text):