Update app.py
Browse files
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),
|
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()
|