Update app/tapas.py

app/tapas.py

@@ -1,7 +1,4 @@
-from transformers import TapasTokenizer, TFTapasForQuestionAnswering
-import pandas as pd
-
-from transformers import TapasTokenizer, TapasForQuestionAnswering
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import pandas as pd
 import re
 
@@ -13,10 +10,12 @@ def load_model_and_tokenizer():
     Load
     """
     # Load pretrained tokenizer: TAPAS finetuned on WikiTable Questions
-    tokenizer = TapasTokenizer.from_pretrained("google/tapas-base-finetuned-wtq")
+    # tokenizer = TapasTokenizer.from_pretrained("google/tapas-base-finetuned-wtq")
+    tokenizer = AutoTokenizer.from_pretrained("Meena/table-question-answering-tapas")
 
     # Load pretrained model: TAPAS finetuned on WikiTable Questions
-    model = TapasForQuestionAnswering.from_pretrained("google/tapas-base-finetuned-wtq")
+    # model = TapasForQuestionAnswering.from_pretrained("google/tapas-base-finetuned-wtq")
+    model = AutoModelForTableQuestionAnswering.from_pretrained("Meena/table-question-answering-tapas")
 
     # Return tokenizer and model
     return tokenizer, model