Update app.py

app.py

@@ -78,7 +78,8 @@ def get_conversation_chain(vector_store):
     #repo_id="HuggingFaceH4/starchat-beta"
     llm = HuggingFaceHub(repo_id=repo_id,
                          model_kwargs={"min_length":1024,
-                                       "max_new_tokens":5632, "do_sample":True,
+                                       #"max_new_tokens":5632, "do_sample":True,
+                                       "max_new_tokens":3072, "do_sample":True,
                                        "temperature":0.1,
                                        "top_k":50,
                                        "top_p":0.95, "eos_token_id":49155}) 
@@ -114,20 +115,23 @@ def main():
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = None
     st.header('Chat with Your own PDFs :books:')
-    question = st.text_input("Ask anything to your PDF: ")
+    
     #if question:
+    vector_store = get_vector_store()
+    st.session_state.conversation =  get_conversation_chain(vector_store)
+    question = st.text_input("Ask anything to your PDF: ")
     if question !="" and not question.strip().isspace() and not question == "" and not question.strip() == "" and not question.isspace():
         handle_user_input(question)  
-    with st.sidebar:
-        st.subheader("Upload your Documents Here: ")
-        pdf_files = st.file_uploader("Choose your PDF Files and Press OK", type=['pdf'], accept_multiple_files=True)
-        if st.button("OK"):
-            with st.spinner("Preparation under process..."):          
-                # Create Vector Store                
-                vector_store = get_vector_store()
-                st.write("DONE")
-                # Create conversation chain
-                st.session_state.conversation =  get_conversation_chain(vector_store)
+#    with st.sidebar:
+#        st.subheader("Upload your Documents Here: ")
+#        pdf_files = st.file_uploader("Choose your PDF Files and Press OK", type=['pdf'], accept_multiple_files=True)
+#        if st.button("OK"):
+#            with st.spinner("Preparation under process..."):          
+#                # Create Vector Store                
+#                vector_store = get_vector_store()
+#                st.write("DONE")
+#                # Create conversation chain
+#                st.session_state.conversation =  get_conversation_chain(vector_store)
 
 if __name__ == '__main__':
     main()