Update app.py

app.py

@@ -35,7 +35,7 @@ def load_pdf_and_generate_embeddings(pdf_doc, open_ai_key):
         Return the key fields from the question followed by : and the answer :"""
         PROMPT = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
         chain_type_kwargs = {"prompt": PROMPT}
-        pdf_qa = RetrievalQA.from_chain_type(llm=ChatOpenAI(temperature=0, model_name="gpt-4"),chain_type="stuff", retriever=vectordb.as_retriever(search_kwargs={"k": 1}), chain_type_kwargs=chain_type_kwargs, return_source_documents=False)
+        pdf_qa = RetrievalQA.from_chain_type(llm=ChatOpenAI(temperature=0, model_name="gpt-4"),chain_type="stuff", retriever=vectordb.as_retriever(), chain_type_kwargs=chain_type_kwargs, return_source_documents=False)
                 
         return "Ready"
     else:
@@ -120,7 +120,7 @@ with gr.Blocks(css=css,theme=gr.themes.Monochrome()) as demo:
         with gr.Row():
             document_type = gr.Radio(['Deed of Trust', 'Transmittal Summary'], label="Select the Document Type")
             answers = gr.Dataframe(label="Answers to Predefined Question set")
-            answers_for_predefined_question_set = gr.Button("Get Answers to Pre-defined Question set").style(full_width=False)
+            answers_for_predefined_question_set = gr.Button("Get gpt-4 answers to pre-defined question set").style(full_width=False)
               
         with gr.Row():
             input = gr.Textbox(label="Type in your question")