Downloads · 30 days
51
3% of all-time downloads
mt5-small-thai-qg
SuperAI2-Machima/mt5-small-thai-qg
mt5-small-thai-qg is a machine learning model from SuperAI2-Machima. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
All-time downloads
1.5K
Public
Repo size
2.4 GB
Likes
4
Public
Repo makeup
Click a slice to open those files.
.bin1.2 GB · 98%
At a glance
- Library
- transformers
- License
- mit
- Model type
- mt5
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Feb 23, 2022
- SHA
- 5e119d28
Try a prompt
Model card
From the Hugging Face model README
SuperAI Engineer Season 2 , Machima
from transformers import T5Tokenizer, T5ForConditionalGeneration, T5Config
model = T5ForConditionalGeneration.from_pretrained('SuperAI2-Machima/mt5-small-thai-qg')
tokenizer = T5Tokenizer.from_pretrained('SuperAI2-Machima/mt5-small-thai-qg')
source_text = 'บุกยึดไม้เถื่อน อดีต ส.ส.บุรีรัมย์ เตรียมสร้างคฤหาสน์ทรงไทย 1 กันยายน 2550 12:00 น. ตำรวจภูธรจ.บุรีรัมย์บุกตรวจยึดไม้แปรรูปหวงห้ามกว่า 80 แผ่น'
print('Predicted Summary Text : ')
tokenized_text = tokenizer.encode(source_text, return_tensors="pt").to(device)
summary_ids = model.generate(tokenized_text,
num_beams=4,
no_repeat_ngram_size=2,
max_length=50,
early_stopping=True)
output = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
print(output)
#Predicted Summary Text :
#answer: 80 แผ่น question: ตํารวจภูธรจ.บุรีรัมย์บุกตรวจยึดไม้แปรรูปหวงห้ามกว่ากี่แผ่น
- Library
- transformers
- Type
- mt5
- License
- mit
- Created
- Mar 2, 2022
- Updated
- Feb 23, 2022