Grey01 commited on
Commit
685e001
·
verified ·
1 Parent(s): ce8de27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ for message in st.session_state.messages:
15
  st.markdown(message["content"])
16
 
17
  # Function for generating responses
18
- def generate_response(message, model, tokenizer):
19
  st.session_state.messages.append({"role": "user", "content": message})
20
  response = client.predict(message=message,api_name="/predict")
21
  st.session_state.messages.append({"role": "assistant", "content": response})
 
15
  st.markdown(message["content"])
16
 
17
  # Function for generating responses
18
+ def generate_response(message):
19
  st.session_state.messages.append({"role": "user", "content": message})
20
  response = client.predict(message=message,api_name="/predict")
21
  st.session_state.messages.append({"role": "assistant", "content": response})