Update app.py

app.py

@@ -27,13 +27,6 @@ def predict(message, system_prompt="", temperature=0.9, max_new_tokens=4096):
 st.title(TITLE)
 st.write(DESCRIPTION)
 
-# Input fields
-message = st.text_area("Enter your message:", "")
-system_prompt = st.text_area("Optional system prompt:", "")
-temperature = st.slider("Temperature", min_value=0.0, max_value=1.0, value=0.9, step=0.05)
-max_new_tokens = st.slider("Max new tokens", min_value=0, max_value=4096, value=4096, step=64)
-
-if st.button("Predict"):
-    response = predict(message, system_prompt, temperature, max_new_tokens)
-    st.write("Response:", response)
-
+prompt = st.chat_input("Say something")
+if prompt:
+    st.write(predict(prompt, '',0.7,4096))