juierror pgzmnk commited on
Commit
5db884f
1 Parent(s): 4124a85

Update README.md (#2)

Browse files

- Update README.md (91ed14cb1c48db588c156e3dbaa7b58101df7448)


Co-authored-by: pgzmnk <pgzmnk@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -11,8 +11,8 @@ widget:
11
  from typing import List
12
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
13
 
14
- tokenizer = AutoTokenizer.from_pretrained("juierror/text-to-sql-with-table-schema")
15
- model = AutoModelForSeq2SeqLM.from_pretrained("juierror/text-to-sql-with-table-schema")
16
 
17
  def prepare_input(question: str, table: List[str]):
18
  table_prefix = "table:"
 
11
  from typing import List
12
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
13
 
14
+ tokenizer = AutoTokenizer.from_pretrained("juierror/flan-t5-text2sql-with-schema")
15
+ model = AutoModelForSeq2SeqLM.from_pretrained("juierror/flan-t5-text2sql-with-schema")
16
 
17
  def prepare_input(question: str, table: List[str]):
18
  table_prefix = "table:"