teaevo commited on
Commit
5bcd1a1
·
1 Parent(s): 6de04aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -86,7 +86,7 @@ sql_model_name = "microsoft/tapex-large-finetuned-wtq"
86
  sql_tokenizer = TapexTokenizer.from_pretrained(sql_model_name)
87
  sql_model = BartForConditionalGeneration.from_pretrained(sql_model_name)
88
 
89
- stokenizer = AutoTokenizer.from_pretrained(model_name)
90
  max_token_limit = stokenizer.max_model_input_sizes[sql_model_name]
91
  print(f"SQL Maximum token limit for {sql_model_name}: {max_token_limit}")
92
 
 
86
  sql_tokenizer = TapexTokenizer.from_pretrained(sql_model_name)
87
  sql_model = BartForConditionalGeneration.from_pretrained(sql_model_name)
88
 
89
+ stokenizer = AutoTokenizer.from_pretrained(sql_model_name)
90
  max_token_limit = stokenizer.max_model_input_sizes[sql_model_name]
91
  print(f"SQL Maximum token limit for {sql_model_name}: {max_token_limit}")
92