alexisjihyeross commited on
Commit
8f68f62
1 Parent(s): 0a59e8d

change model path

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -20,8 +20,7 @@ For more code release, please refer to [this github page](https://github.com/all
20
  ```python
21
  from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
22
 
23
- # TODO: change to the correct name
24
- model_path = "uw-hai/tailor_private"
25
  generator = pipeline("text2text-generation",
26
  model=AutoModelForSeq2SeqLM.from_pretrained(model_path),
27
  tokenizer=AutoTokenizer.from_pretrained(model_path),
 
20
  ```python
21
  from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
22
 
23
+ model_path = "allenai/tailor"
 
24
  generator = pipeline("text2text-generation",
25
  model=AutoModelForSeq2SeqLM.from_pretrained(model_path),
26
  tokenizer=AutoTokenizer.from_pretrained(model_path),