Update app/tapex.py

app/tapex.py

@@ -8,7 +8,6 @@ def execute_query(query, csv_file):
     a = datetime.datetime.now()
 
     table = pd.read_csv(csv_file.name, delimiter=",")
-    table.fillna(0, inplace=True)
     table = table.astype(str)
 
     model_name = "microsoft/tapex-large-finetuned-wtq"
@@ -17,7 +16,7 @@ def execute_query(query, csv_file):
 
     queries = [query]
 
-    encoding = tokenizer(table=table, queries=queries, padding=True, return_tensors="pt",truncation=True)
+    encoding = tokenizer(table=table, query=queries, padding=True, return_tensors="pt",truncation=True)
     outputs = model.generate(**encoding)
     ans = tokenizer.batch_decode(outputs, skip_special_tokens=True)