Update app.py

app.py

@@ -27,6 +27,9 @@ def predict(message, system_prompt="", temperature=0.9, max_new_tokens=4096):
 st.title(TITLE)
 st.write(DESCRIPTION)
 
-prompt = st.chat_input("Say something")
+prompt = st.chat_input("Ask llama 2...")
 if prompt:
-    st.write(predict(prompt, '',0.7,4096))
+    with st.chat_message("user"):
+        st.write(prompt)
+    with st.chat_message("llama",'🦙'):
+        st.write(predict(prompt, '',0.7,4096))