Skip to content

Latest commit

 

History

History

t5_base_japanese_title_generation

T5 Base Japanese Title Generation

Input

TEXT file.

Output

The title appropriate to the input prompt.

aiの認識力は飛躍的に向上!大量のデータから学習された基盤モデル

Usage

Automatically downloads the onnx and prototxt files on the first run. It is necessary to be connected to the Internet while downloading.

For the sample text,

$ python3 t5_base_japanese_title_generation.py

If you want to specify the text or pdf file, put the file path after the -i option.

$ python3 t5_base_japanese_title_generation.py -i FILE_PATH

Reference

Framework

PyTorch

Model Format

ONNX opset=12

Netron

encoder

t5-base-japanese-title-generation-encoder.onnx.prototxt

decoder

t5-base-japanese-title-generation-decoder.onnx.prototxt