teaevo commited on
Commit
9d8ae89
·
1 Parent(s): 2ce1434

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -105,7 +105,7 @@ def sqlquery(input, history=[]):
105
  #sql_response = sql_tokenizer.batch_decode(sql_outputs, skip_special_tokens=True)
106
 
107
  # append the new user input tokens to the chat history
108
- bot_input_ids = torch.cat([torch.LongTensor(history), sql_encoding], dim=-1)
109
 
110
  # generate a response
111
  history = sql_model.generate(bot_input_ids, max_length=1000, pad_token_id=sql_tokenizer.eos_token_id).tolist()
 
105
  #sql_response = sql_tokenizer.batch_decode(sql_outputs, skip_special_tokens=True)
106
 
107
  # append the new user input tokens to the chat history
108
+ bot_input_ids = torch.cat([torch.LongTensor(history), **sql_encoding], dim=-1)
109
 
110
  # generate a response
111
  history = sql_model.generate(bot_input_ids, max_length=1000, pad_token_id=sql_tokenizer.eos_token_id).tolist()