Skip to content

Latest commit

 

History

History

t5_base_japanese_summarization

T5 base Japanese summarization

Abstract summarization model made by fine-tuning sonoisa/t5-base-japanese-title-generation

Input

TEXT file. This text was adapted from the wikipedia page of Honfukuji temple.

Output

Abstract summarization of input file.

淡路島にある本福寺は、本堂と回廊で結ばれている。蓮池の周囲を回ると、大きなアーチ状の大扉が現れる。蓮池の下には極楽浄土が出現するという演出がなされている。

Usage

Internet connection is required when running the script for the first time, as the model files will be automatically downloaded.

Summarization of the text in ./input.txt will be automatically generated by excecuting the script below.

$ python3 t5_base_japanese_summarization.py

Use argument '-i' to input a text directly to the script, and use '-f' to specify a file to be the input.

Reference

Framework

PyTorch

Model Format

ONNX opset=12

Netron

encoder

t5_base_japanese_summarization_enc.onnx.prototxt

decoder

t5_base_japanese_summarization_dec.onnx.prototxt