PD03 commited on
Commit
03083eb
·
verified ·
1 Parent(s): 475c7b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ schema = ", ".join(df.columns) # e.g. "Region, Product, FiscalYear, ..."
18
  MODEL_ID = "mrm8488/t5-base-finetuned-wikisql"
19
  device = 0 if torch.cuda.is_available() else -1
20
 
21
- tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
22
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_ID)
23
 
24
  sql_generator = pipeline(
 
18
  MODEL_ID = "mrm8488/t5-base-finetuned-wikisql"
19
  device = 0 if torch.cuda.is_available() else -1
20
 
21
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, use_fast=False)
22
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_ID)
23
 
24
  sql_generator = pipeline(