Update app.py

app.py

@@ -49,9 +49,9 @@ def predict(input, history=[]):
     bot_input_ids = torch.cat([torch.LongTensor(history), new_user_input_ids], dim=-1)
 
     if is_question: 
-        sql_encoding = sql_tokenizer(table=table, query=user_message, return_tensors="pt")
-        sql_outputs = sql_model.generate(**encoding)
-        sql_response = sql_tokenizer.batch_decode(outputs, skip_special_tokens=True)
+        sql_encoding = sql_tokenizer(table=table, query=input, return_tensors="pt")
+        sql_outputs = sql_model.generate(**sql_encoding)
+        sql_response = sql_tokenizer.batch_decode(sql_outputs, skip_special_tokens=True)
         
     # generate a response
     history = model.generate(bot_input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id).tolist()