DrDomedag commited on
Commit
a843d7e
·
1 Parent(s): 8e03e52

Initial commit.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ if st.button("Send"):
31
 
32
  # Query the model
33
  with st.spinner("Zephyr is thinking..."):
34
- response = client.text_generation(user_input, parameters={"max_new_tokens": 100})
35
  assistant_reply = response.get("generated_text", "").strip()
36
 
37
  # Add the model's reply to the chat history
 
31
 
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