Update app.py

app.py

@@ -104,8 +104,9 @@ def sqlquery(input, history=[]):
     #sql_outputs = sql_model.generate(**sql_encoding)
     #sql_response = sql_tokenizer.batch_decode(sql_outputs, skip_special_tokens=True)
 
+    sql_input_ids = sql_encoding["input_ids"]
      # append the new user input tokens to the chat history
-    bot_input_ids = torch.cat([torch.LongTensor(history),  **sql_encoding], dim=-1)
+    bot_input_ids = torch.cat([torch.LongTensor(history),  sql_input_ids], dim=-1)
       
     # generate a response
     history = sql_model.generate(bot_input_ids, max_length=1000, pad_token_id=sql_tokenizer.eos_token_id).tolist()