DrDomedag commited on
Commit
3a8a5cf
·
1 Parent(s): a843d7e

Initial commit.

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -32,7 +32,8 @@ if st.button("Send"):
32
  # Query the model
33
  with st.spinner("Zephyr is thinking..."):
34
  response = client.text_generation(user_input)
35
- assistant_reply = response.get("generated_text", "").strip()
 
36
 
37
  # Add the model's reply to the chat history
38
  st.session_state.messages.append({"role": "assistant", "content": assistant_reply})
 
32
  # Query the model
33
  with st.spinner("Zephyr is thinking..."):
34
  response = client.text_generation(user_input)
35
+ #assistant_reply = response.get("generated_text", "").strip()
36
+ assistant_reply = response.strip()
37
 
38
  # Add the model's reply to the chat history
39
  st.session_state.messages.append({"role": "assistant", "content": assistant_reply})