fix model selection

app.py

@@ -104,9 +104,15 @@ with st.sidebar:
             use_expert_answer = st.toggle("Use expert answer", value=True)
             show_expert_responce = st.toggle("Show initial expert answer", value=False)
 
-            model = st.selectbox("Choose the LLM model", ["gpt-4o-mini", "gpt-3.5-turbo", "llama-tommi-0.35"], key='a1model')
+            st.session_state.expert_model = st.selectbox(
+                "Choose the LLM model",
+                ["gpt-4o-mini",
+                 "gpt-3.5-turbo",
+                 "llama-tommi-0.35"],
+                key='a1model'
+            )
 
-            if model == "llama-tommi-0.35":
+            if st.session_state.expert_model == "llama-tommi-0.35":
                 tommi_do_sample = st.toggle("Enable Sampling", value=False, key='tommi_sample')
 
                 if tommi_do_sample:
@@ -220,7 +226,7 @@ if submit_button_placeholder.button("AI Answer", type="primary"):
                     context += context_item['text'] + '\n\n'
             
             if use_expert_answer:
-                if model == "llama-tommi-0.35":
+                if st.session_state.expert_model  == "llama-tommi-0.35":
                     if 'tommi_model' not in st.session_state:
                         tommi_model, tommi_tokenizer = load_fine_tuned_model(adapter_path, base_model_path)
                         st.session_state.tommi_model = tommi_model
@@ -242,7 +248,7 @@ if submit_button_placeholder.button("AI Answer", type="primary"):
                         num_beams=tommi_num_beams if not tommi_do_sample else 1,
                         max_new_tokens=tommi_max_new_tokens
                     )
-                elif model in ["gpt-4o-mini", "gpt-3.5-turbo"]:
+                elif st.session_state.expert_model  in ["gpt-4o-mini", "gpt-3.5-turbo"]:
                     expert_answer = openai_domain_specific_answer_generation(
                         get_expert_system_prompt(),
                         st.session_state.question,