Update app.py

app.py

@@ -14,7 +14,6 @@ import textwrap
 def main():
     st.set_page_config(page_title="Talk with BORROWER data")
     st.title("Talk with BORROWER data")
-    uploaded_file = st.sidebar.file_uploader("Upload your Data", type="csv")
 
     query = st.text_input("Send a Message")
     if st.button("Submit Query", type="primary"):
@@ -24,7 +23,6 @@ def main():
         loader = CSVLoader(file_path="./borrower_data.csv", encoding="utf-8", csv_args={
                     'delimiter': ','})
         data = loader.load()
-        st.write(data)
 
         model = "daryl149/llama-2-7b-chat-hf" 
         tokenizer = AutoTokenizer.from_pretrained(model) 
@@ -35,7 +33,7 @@ def main():
                                           trust_remote_code=True, 
                                           device_map="auto",  
                                           do_sample=True, 
-                                          top_k=5, 
+                                          top_k=1, 
                                           num_return_sequences=1, 
                                           eos_token_id=tokenizer.eos_token_id 
         )