fix: form

app.py

@@ -122,6 +122,7 @@ def init_session_state():
     st.session_state["repetition_penalty"] = 1.1
     st.session_state["system_prompt"] = "You are a medical expert that provides answers for a medically trained audience"
     st.session_state["prompt"] = ""
+    st.session_state["llm_messages"] = []
 
 def get_genders():
     return ['Male', 'Female']
@@ -225,7 +226,7 @@ def display_llm_output():
         formatted_prompt = format_prompt(prompt_format, system_prompt, prompt)
         print(f"Formatted prompt: {format_prompt}")
         llm_response = get_llm_response(
-            st.session_state["model_name"],
+            st.session_state["model_name_or_path"],
             st.session_state["temperature"],
             st.session_state["do_sample"],
             st.session_state["top_p"],