teaevo commited on
Commit
d69d5ee
·
1 Parent(s): c17ba77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,8 +25,8 @@ chatbot_model = AutoModelForCausalLM.from_pretrained(chatbot_model_name)
25
  #model_name = "microsoft/tapex-large-finetuned-wikisql" # You can change this to any other model from the list above
26
  #model_name = "microsoft/tapex-base-finetuned-wikisql"
27
  #model_name = "microsoft/tapex-base-finetuned-wtq"
28
- #model_name = "microsoft/tapex-large-finetuned-wtq"
29
- model_name = "google/tapas-base-finetuned-wtq"
30
  sql_tokenizer = TapexTokenizer.from_pretrained(model_name)
31
  sql_model = BartForConditionalGeneration.from_pretrained(model_name)
32
 
 
25
  #model_name = "microsoft/tapex-large-finetuned-wikisql" # You can change this to any other model from the list above
26
  #model_name = "microsoft/tapex-base-finetuned-wikisql"
27
  #model_name = "microsoft/tapex-base-finetuned-wtq"
28
+ model_name = "microsoft/tapex-large-finetuned-wtq"
29
+ #model_name = "google/tapas-base-finetuned-wtq"
30
  sql_tokenizer = TapexTokenizer.from_pretrained(model_name)
31
  sql_model = BartForConditionalGeneration.from_pretrained(model_name)
32