umisato's picture
Update README.md
e350a07 verified
|
raw
history blame
577 Bytes
metadata
library_name: transformers
tags:
  - text2text-generation
pipeline_tag: text2text-generation

Your Model Name

This model is designed for text generation tasks.

Example Usage

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("umisato/mt5-base-kirakira-names")
model = AutoModelForSeq2SeqLM.from_pretrained("umisato/mt5-base-kirakira-names")

inputs = tokenizer("your-input(kanji)", return_tensors="pt")
outputs = model.generate(**inputs)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))