Update app.py

app.py

@@ -64,11 +64,16 @@ def get_answer(chatbot, input_text):
     output += f"\nAnswered in {elapsed_time:.1f} seconds, Nr generated words: {count_words(output)}"
     
     return output
-    
+ 
+if "model_name" not in st.session_state.keys():
+    # Initialize the model with the default option
+    st.session_state["model_name"] = "BramVanroy/Llama-2-13b-chat-dutch"
+    my_config.update({'model_name': st.session_state["model_name"]})
+    llm_chatbot = generate_with_llama_dutch(my_config)
+    st.session_state["model"] = llm_chatbot   
 
-chatbot = generate_with_llama_chat(my_config)
 text = st.text_area("Enter text to summarize here.")
 
 if text:
-    out = get_answer(chatbot, text)
+    out = get_answer(llm_chatbot, text)
     st.write(out)