Spaces:

patrawtf
/

shopify_csv_qa

Runtime error

patrawtf commited on Mar 17, 2023

Commit

ef21cbc

1 Parent(s): 3720e70

Update app/tapex.py

Files changed (1) hide show

app/tapex.py CHANGED Viewed

@@ -17,7 +17,7 @@ def execute_query(query, csv_file):
     queries = [query]
-    encoding = tokenizer(table=table, queries=queries, padding=True, return_tensors="pt",truncated=True)
     outputs = model.generate(**encoding)
     ans = tokenizer.batch_decode(outputs, skip_special_tokens=True)

     queries = [query]
+    encoding = tokenizer(table=table, queries=queries, padding=True, return_tensors="pt",truncation=True)
     outputs = model.generate(**encoding)
     ans = tokenizer.batch_decode(outputs, skip_special_tokens=True)