Stevross commited on
Commit
50a1c09
1 Parent(s): bd6d8db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ user_input = st.text_input("You: ", "")
14
  # Get response from the chatbot
15
  if st.button("Ask"):
16
  with st.spinner("Generating response..."):
17
- response = chatbot(user_input, max_length=100, do_sample=True, top_p=0.95, top_k=60)
18
  st.write("Bot:", response[0]['generated_text'])
19
 
20
  st.sidebar.header("About")
 
14
  # Get response from the chatbot
15
  if st.button("Ask"):
16
  with st.spinner("Generating response..."):
17
+ response = chatbot(user_input, max_length=100, do_sample=True, top_p=0.95, top_k=60, trust_remote_code=True)
18
  st.write("Bot:", response[0]['generated_text'])
19
 
20
  st.sidebar.header("About")