Rename app/tapas.py to app/tapex.py

app/{tapas.py → tapex.py}

@@ -1,6 +1,7 @@
 from transformers import TapasTokenizer, TapexTokenizer, BartForConditionalGeneration
 import pandas as pd
 import datetime
+import torch
 
 
 def execute_query(query, csv_file):
@@ -16,7 +17,7 @@ def execute_query(query, csv_file):
 
     queries = [query]
 
-    encoding = tokenizer(table=table, queries=queries, padding=True, return_tensors="tf",truncated=True)
+    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)