Update app.py

app.py

@@ -42,10 +42,19 @@ message_input = st.text_input("You:", "")
 system_prompt_input = st.text_input("System Prompt:", "You are a helpful assistant.")
 
 # Create sliders for temperature, max new tokens, top-p, and repetition penalty
-temperature_slider = st.slider("Temperature", 0.0, 1.0, 0.9)
-max_new_tokens_slider = st.slider("Max new tokens", 0, 1048, 256)
-top_p_slider = st.slider("Top-p (nucleus sampling)", 0.0, 1.0, 0.95)
-repetition_penalty_slider = st.slider("Repetition penalty", 1.0, 2.0, 1.0)
+temperature_slider = st.slider("Temperature", 0.0, 1.0, 0.9, key="temperature_slider")
+max_new_tokens_slider = st.slider("Max new tokens", 0, 1048, 256, key="max_new_tokens_slider")
+top_p_slider = st.slider("Top-p (nucleus sampling)", 0.0, 1.0, 0.95, key="top_p_slider")
+repetition_penalty_slider = st.slider("Repetition penalty", 1.0, 2.0, 1.0, key="repetition_penalty_slider")
+
+# Create a container for the sliders
+slider_container = st.container()
+
+# Add the sliders to the container
+slider_container.columns([1, 1, 1, 1])[0].slider("Temperature", 0.0, 1.0, 0.9, key="temperature_slider_left")
+slider_container.columns([1, 1, 1, 1])[1].slider("Max new tokens", 0, 1048, 256, key="max_new_tokens_slider_left")
+slider_container.columns([1, 1, 1, 1])[2].slider("Top-p (nucleus sampling)", 0.0, 1.0, 0.95, key="top_p_slider_left")
+slider_container.columns([1, 1, 1, 1])[3].slider("Repetition penalty", 1.0, 2.0, 1.0, key="repetition_penalty_slider_left")
 
 # Create button to generate response
 if st.button("Generate"):