Update app.py

app.py

@@ -17,7 +17,7 @@ from langchain_community.llms import HuggingFaceEndpoint
 import torch
 api_token = os.getenv("HF_TOKEN")
 
-list_llm = ["meta-llama/Meta-Llama-3-8B-Instruct", "mistralai/Mistral-7B-Instruct"]
+list_llm = ["meta-llama/Meta-Llama-3-8B-Instruct", "mistralai/Mistral-7B-Instruct-v0.3"]
 list_llm_simple = [os.path.basename(llm) for llm in list_llm]
 
 # Load PDF document and create doc splits
@@ -174,7 +174,7 @@ def demo():
         vector_db = gr.State()
         qa_chain = gr.State()
         collection_name = gr.State()
-        initial_prompt = gr.State()
+        initial_prompt = gr.State("")
         llm_no_doc = gr.State()
 
         gr.Markdown(
@@ -254,7 +254,7 @@ def demo():
         db_btn.click(initialize_database, 
             inputs=[document, slider_chunk_size, slider_chunk_overlap, db_type_radio], 
             outputs=[vector_db, collection_name, db_progress])
-        set_prompt_btn.click(lambda prompt: prompt, 
+        set_prompt_btn.click(lambda prompt: gr.State(prompt), 
                              inputs=prompt_input, 
                              outputs=initial_prompt)
         qachain_btn.click(initialize_LLM,