metadata

library_name: transformers
tags:
  - text2text-generation
pipeline_tag: text2text-generation

Your Model Name

このモデルはキラキラネームの読み方を予測するために、mT5モデルをファインチューニングをしたものです。読み方を予測するため、入力は漢字である必要があります。

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))