Update app.py

app.py

@@ -17,7 +17,7 @@ with st.sidebar:
     st.markdown("### Instructions 🤖\nThis is a basic chatbot. Ask anything, and the AI will try to help you! The app is supported by Yiqiao Yin.")
 
     # Add a section to ask the user for the response length
-    st.text("Select the desired length of the AI response:")
+    st.markdown("#### Select the desired length of the AI response:")
     response_length = st.radio(
         "How detailed do you want the response to be?",
         ('Efficient', 'Medium', 'Academic')
@@ -37,8 +37,8 @@ with st.sidebar:
     st.write(f"Copyright © 2010-{datetime.now().year} Present Yiqiao Yin")
 
 # Initialize session state variables if not already present
-if "openai_model" not in st.session_state:
-    st.session_state["openai_model"] = "nvidia/llama-3.1-nemotron-70b-instruct"
+if "nvidia_model" not in st.session_state:
+    st.session_state["nvidia_model"] = "nvidia/llama-3.1-nemotron-70b-instruct"
 
 if "messages" not in st.session_state:
     # Adding the initial system message
@@ -61,7 +61,7 @@ if prompt := st.chat_input("What is up?"):
         with st.spinner("The assistant is thinking... Please wait."):
             # Create Nvidia completion request with full conversation history
             stream = client.chat.completions.create(
-                model=st.session_state["openai_model"],
+                model=st.session_state["nvidia_model"],
                 messages=st.session_state.messages,  # Include all previous messages in the API call
                 temperature=0.5,
                 top_p=0.7,