Initial commit.
Browse files
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})
|