Update app.py

app.py

@@ -66,15 +66,15 @@ def main():
             #PATH = 'model/'
             #llm = AutoModelForCausalLM.from_pretrained("CohereForAI/aya-101")
             # llm = AutoModelForCausalLM.from_pretrained(PATH,local_files_only=True)
-            llm = huggingface_hub.HuggingFaceHub(repo_id="google/flan-t5-small",
+            llm = huggingface_hub.HuggingFaceHub(repo_id="google/flan-t5-large",
                                                   model_kwargs={"temperature":1.0, "max_length":256})
             if query == 'Summarize':
                 #docs = pdf_reader.load_and_split()
                 chain = load_summarize_chain(llm, chain_type="map_reduce")
                 response = chain.run(chunks_doc[0])
             else:
-                docs = vector_store.similarity_search(query=query, k=2)
-                #st.write(docs)
+                docs = vector_store.similarity_search(query=query, k=3)
+                st.write(docs)
                 chain = load_qa_chain(llm=llm, chain_type="stuff")
                 response = chain.run(input_documents=docs, question=query)
                 #retriever=vector_store.as_retriever()