pythainlp/thaisum
Viewer • Updated • 392k • 158 • 4
How to use EXt1/KMUTT_CPE35_thai-mt5base_summarizer-V2 with Transformers:
# Use a pipeline as a high-level helper
# Warning: Pipeline type "summarization" is no longer supported in transformers v5.
# You must load the model directly (see below) or downgrade to v4.x with:
# 'pip install "transformers<5.0.0'
from transformers import pipeline
pipe = pipeline("summarization", model="EXt1/KMUTT_CPE35_thai-mt5base_summarizer-V2") # Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("EXt1/KMUTT_CPE35_thai-mt5base_summarizer-V2", device_map="auto")This repository contains a fine-tuned version of google/mt5-base for the task of Thai text summarization. The model was trained on 50,000 samples from the ThaiSum dataset and is part of a senior project in the Computer Engineering Department at King Mongkut’s University of Technology Thonburi (KMUTT).
google/mt5-baseThe performance of the model was evaluated using the ROUGE metric, which is commonly used for assessing the quality of summarization tasks. The evaluation results on the test set are as follows:
Base model
google/mt5-base