Update app.py
Browse files
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=
|
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:
|