Update Demo.py

Demo.py

@@ -53,17 +53,12 @@ def create_pipeline(model):
         .setInputCols(["document_table"])\
         .setOutputCol("table")
 
-    tapas_wtq = TapasForQuestionAnswering\
-        .pretrained("table_qa_tapas_base_finetuned_wtq", "en")\
+    tapas = TapasForQuestionAnswering\
+        .pretrained(model, "en")\
         .setInputCols(["questions", "table"])\
         .setOutputCol("answers_wtq")
-    
-    tapas_sqa = TapasForQuestionAnswering\
-        .pretrained("table_qa_tapas_base_finetuned_sqa", "en")\
-        .setInputCols(["questions", "table"])\
-        .setOutputCol("answers_sqa")
 
-    pipeline = Pipeline(stages=[document_assembler, sentence_detector, table_assembler, tapas_wtq, tapas_sqa])
+    pipeline = Pipeline(stages=[document_assembler, sentence_detector, table_assembler, tapas])
     return pipeline
 
 def fit_data(pipeline, json_data, question):