rushic24 commited on
Commit
11a7380
1 Parent(s): 3893ca4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ if 'past' not in st.session_state:
24
  st.session_state['past'] = []
25
 
26
  def query(payload):
27
- bot_input_ids = tokenizer.encode(payload["text"] + tokenizer.eos_token, return_tensors='pt')
28
 
29
  chat_history_ids = model.generate(
30
  bot_input_ids, max_length=100,
 
24
  st.session_state['past'] = []
25
 
26
  def query(payload):
27
+ bot_input_ids = tokenizer.encode(payload["inputs"]["text"] + tokenizer.eos_token, return_tensors='pt')
28
 
29
  chat_history_ids = model.generate(
30
  bot_input_ids, max_length=100,