Update pages/Web-UI-ld.py

pages/Web-UI-ld.py

@@ -2,6 +2,10 @@ import streamlit as st
 from streamlit_chat import message
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
+# Initialize the model_name key in the st.session_state dictionary
+if 'model_name' not in st.session_state:
+    st.session_state['model_name'] = "microsoft/DialoGPT-medium"
+
 # Define the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained(st.session_state['model_name'])
 model = AutoModelForCausalLM.from_pretrained(st.session_state['model_name'])
@@ -10,8 +14,6 @@ if 'conversation' not in st.session_state:
     st.session_state['conversation'] = None
 if 'messages' not in st.session_state:
     st.session_state['messages'] = []
-if 'model_name' not in st.session_state:
-    st.session_state['model_name'] = "microsoft/DialoGPT-medium"
 
 # Setting page title and header
 st.set_page_config(page_title="Chat GPT Clone", page_icon=":robot_face:")
@@ -19,7 +21,7 @@ st.markdown("<h1 style='text-align: center;'>💻 ChatterBot</h1>", unsafe_allow
 st.subheader("How Can I Help You Today? 🤖")
 
 st.sidebar.title("😎🗝️")
-st.session_state['model_name'] = st.sidebar.text_input("Current Model", value="microsoft/DialoGPT-medium")
+st.session_state['model_name'] = st.sidebar.text_input("Hugging Face Model Name", value="microsoft/DialoGPT-medium")
 summarise_button = st.sidebar.button("Summarise the conversation", key="summarise")
 st.sidebar.image('./chatbot.jpg', width=300, use_column_width=True)
 if summarise_button: