Update app.py

app.py

@@ -45,7 +45,8 @@ models = {
 }
 
 # Layout for model selection and max_tokens slider
-col1, col2 = st.columns(2)
+col1, col2 = st.columns([1, 3]) # Adjust the ratio to make the first column smaller
+
 
 with col1:
  model_option = st.selectbox(
@@ -54,26 +55,21 @@ with col1:
  format_func=lambda x: models[x]["name"],
  index=0, # Default to the first model in the list
  )
-
-# Detect model change and clear chat history if model has changed
-if st.session_state.selected_model != model_option:
- st.session_state.messages = []
- st.session_state.selected_model = model_option
-
-max_tokens_range = models[model_option]["tokens"]
-
-with col2:
- # Adjust max_tokens slider dynamically based on the selected model
+ max_tokens_range = models[model_option]["tokens"]
  max_tokens = st.slider(
  "Max Tokens:",
- min_value=512, # Minimum value to allow some flexibility
+ min_value=512,
  max_value=max_tokens_range,
- # Default value or max allowed if less
  value=min(32768, max_tokens_range),
  step=512,
  help=f"Adjust the maximum number of tokens (words) for the model's response. Max for selected model: {max_tokens_range}",
  )
 
+# Detect model change and clear chat history if model has changed
+if st.session_state.selected_model != model_option:
+ st.session_state.messages = []
+ st.session_state.selected_model = model_option
+
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
  avatar = "🤖" if message["role"] == "assistant" else "🕺"