fschwartzer commited on
Commit
398cdbe
1 Parent(s): 6d02d8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -22,9 +22,10 @@ def response(user_question, table_data):
22
  #tokenizer = TapexTokenizer.from_pretrained(model_name)
23
  tokenizer = AutoTokenizer.from_pretrained(model_name)
24
 
25
- queries = [user_question]
26
 
27
- encoding = tokenizer(table=table_data, query=queries, padding=True, return_tensors="pt", truncation=True)
 
28
 
29
  # Experiment with generation parameters
30
  outputs = model.generate(
 
22
  #tokenizer = TapexTokenizer.from_pretrained(model_name)
23
  tokenizer = AutoTokenizer.from_pretrained(model_name)
24
 
25
+ #queries = [user_question]
26
 
27
+ #encoding = tokenizer(table=table_data, query=queries, padding=True, return_tensors="pt", truncation=True)
28
+ encoding = tokenizer(table=table_data, query=user_question)
29
 
30
  # Experiment with generation parameters
31
  outputs = model.generate(