Update app.py

app.py

@@ -2,9 +2,13 @@ import gradio as gr
 from transformers import TapexTokenizer, BartForConditionalGeneration
 import pandas as pd
 
+#wikisql take longer to process
 #model_name = "microsoft/tapex-large-finetuned-wikisql"  # You can change this to any other model from the list above
+#model_name = "microsoft/tapex-base-finetuned-wikisql"
+
 #model_name = "microsoft/tapex-large-finetuned-wtq"
-model_name = "microsoft/tapex-base-finetuned-wikisql"
+model_name = "microsoft/tapex-base-finetuned-wtq"
+
 tokenizer = TapexTokenizer.from_pretrained(model_name)
 model = BartForConditionalGeneration.from_pretrained(model_name)