Update app.py

app.py

@@ -66,18 +66,13 @@ def main():
             #PATH = 'model/'
             #llm = AutoModelForCausalLM.from_pretrained("openai-community/gpt2-medium")
             llm = huggingface_hub.HuggingFaceHub(repo_id="openai-community/gpt2-medium",model_kwargs={"temperature":1.0, "max_length":500})
-            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=1)
-                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()
-                #chain = RetrievalQA.from_chain_type(llm=llm,chain_type="stuff",retriever=retriever)
-                #response = chain.run(chain)
+            docs = vector_store.similarity_search(query=query, k=1)
+            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()
+            chain = RetrievalQA.from_chain_type(llm=llm,chain_type="stuff",retriever=retriever)
+            response = chain.run(chain)
             st.write(response)