teaevo commited on
Commit
ffc0ad6
·
1 Parent(s): 69beb29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,8 +41,8 @@ def chatbot_response(user_message):
41
 
42
  if is_question:
43
  # If the user input is a question, use TAPEx for question-answering
44
- inputs = user_query
45
- encoding = sql_tokenizer(table=table, query=inputs, return_tensors="pt")
46
  outputs = sql_model.generate(**encoding)
47
  response = sql_tokenizer.batch_decode(outputs, skip_special_tokens=True)
48
  else:
 
41
 
42
  if is_question:
43
  # If the user input is a question, use TAPEx for question-answering
44
+ #inputs = user_query
45
+ encoding = sql_tokenizer(table=table, query=user_message, return_tensors="pt")
46
  outputs = sql_model.generate(**encoding)
47
  response = sql_tokenizer.batch_decode(outputs, skip_special_tokens=True)
48
  else: