Update app.py

app.py

@@ -66,8 +66,8 @@ def main():
             #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",
-                                                  model_kwargs={"temperature":1.0, "max_length":256})
-            docs = vector_store.similarity_search(query=query, k=1)
+                                                  model_kwargs={"temperature":1.0, "max_length":500})
+            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)