Update app.py

app.py

@@ -15,17 +15,12 @@ print(table_data.head())
 def response(user_question, table_data):
     a = datetime.datetime.now()
 
-    #model_name = "microsoft/tapex-large-finetuned-wtq"
     model_name = "google/tapas-base-finetuned-wtq"
-    #model = BartForConditionalGeneration.from_pretrained(model_name)
     model = AutoModelForTableQuestionAnswering.from_pretrained(model_name)
-    #tokenizer = TapexTokenizer.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    
-    #queries = [user_question]
 
-    #encoding = tokenizer(table=table_data, query=queries, padding=True, return_tensors="pt", truncation=True)
-    encoding = tokenizer(table=table_data, query=user_question)
+    # The query should be passed as a list
+    encoding = tokenizer(table=table_data, queries=[user_question], padding=True, return_tensors="pt", truncation=True)
 
     # Experiment with generation parameters
     outputs = model.generate(