umisato commited on
Commit
e350a07
1 Parent(s): 2467064

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -7
README.md CHANGED
@@ -1,9 +1,8 @@
1
  ---
2
  library_name: transformers
3
  tags:
4
- - text-generation
5
- - causal-lm
6
- pipeline_tag: text-generation
7
  ---
8
 
9
  # Your Model Name
@@ -13,11 +12,11 @@ This model is designed for text generation tasks.
13
  ## Example Usage
14
 
15
  ```python
16
- from transformers import AutoTokenizer, AutoModelForCausalLM
17
 
18
- tokenizer = AutoTokenizer.from_pretrained("your-username/your-model-repo")
19
- model = AutoModelForCausalLM.from_pretrained("your-username/your-model-repo")
20
 
21
- inputs = tokenizer("Hello, how are you?", return_tensors="pt")
22
  outputs = model.generate(**inputs)
23
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 
1
  ---
2
  library_name: transformers
3
  tags:
4
+ - text2text-generation
5
+ pipeline_tag: text2text-generation
 
6
  ---
7
 
8
  # Your Model Name
 
12
  ## Example Usage
13
 
14
  ```python
15
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
16
 
17
+ tokenizer = AutoTokenizer.from_pretrained("umisato/mt5-base-kirakira-names")
18
+ model = AutoModelForSeq2SeqLM.from_pretrained("umisato/mt5-base-kirakira-names")
19
 
20
+ inputs = tokenizer("your-input(kanji)", return_tensors="pt")
21
  outputs = model.generate(**inputs)
22
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))