Update app.py

app.py

@@ -27,10 +27,8 @@ def llm_pipeline():
         'text2text-generation',
         model = base_model,
         tokenizer = tokenizer,
-        max_length = 226,
-        do_sample = True,
-        temperature = 0.5,
-        top_p= 0.95
+        max_length = 500,
+        temperature = 0.5
     )
     local_llm = HuggingFacePipeline(pipeline=pipe)
     return local_llm
@@ -74,7 +72,7 @@ def main():
             """
         )
 
-    user_input = st.text_input("", key="input")
+  d =  """ user_input = st.text_input("", key="input")
 
      # Initialize session state for generated responses and past messages
     if "generated" not in st.session_state:
@@ -91,9 +89,8 @@ def main():
 
     # Display conversation history using Streamlit messages
     if st.session_state["generated"]:
-        display_conversation(st.session_state)
+        display_conversation(st.session_state)"""
 
-d = """
     user_input = st.text_input("Question:", placeholder="Ask about your PDF", key='input')
     with st.form(key='my_form', clear_on_submit=True):
         submit_button = st.form_submit_button(label='Send')
@@ -112,7 +109,7 @@ d = """
         st.session_state['generated'].append(answer)
 
     if st.session_state["generated"]:
-       display_conversation(st.session_state)"""
+       display_conversation(st.session_state)
 
   
 if __name__ == '__main__':