Illia56 commited on
Commit
ffd3581
1 Parent(s): 6761ae5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ st.write(DESCRIPTION)
29
 
30
  prompt = st.chat_input("Ask llama 2...")
31
  if prompt:
32
- with st.chat_message("user"):
33
  st.write(prompt)
34
  with st.chat_message("llama", avatar='🦙'):
35
  st.write(predict(prompt, '',0.7,4096))
 
29
 
30
  prompt = st.chat_input("Ask llama 2...")
31
  if prompt:
32
+ with st.chat_message("assistant"):
33
  st.write(prompt)
34
  with st.chat_message("llama", avatar='🦙'):
35
  st.write(predict(prompt, '',0.7,4096))