模型:
thanathorn/mt5-cpe-kmutt-thai-sentence-sum
该代码库包含了用于泰语句子摘要的精调mT5-base模型。该模型的架构基于mT5模型,并在泰语文本摘要对上进行了精调。此外,这个项目是曼谷工业大学计算机工程学生的毕业项目。
from simpletransformers.t5 import T5Model, T5Args from torch import cuda model = T5Model("t5", "thanathorn/mt5-cpe-kmutt-thai-sentence-sum", use_cuda=cuda.is_available()) sentence = "simplify: ถ้าพูดถึงขนมหวานในตำนานที่ชื่นใจที่สุดแล้วละก็ต้องไม่พ้น น้ำแข็งใส แน่เพราะว่าเป็นอะไรที่ชื่นใจสุด" prediction = model.predict([sentence]) print(prediction[0])
(请参考 Google Colab 上的示例)