File size: 268 Bytes
9c50174
 
a9eff47
fa0655e
a9eff47
fa0655e
a9eff47
fa0655e
a9eff47
1
2
3
4
5
6
7
8
9
# Generate News in Thai language by keywords.

MODEL_NAME = 'nonamenlp/news_gen' 

TOKENIZER_NAME = "nonamenlp/news_gen"

trained_model = MT5ForConditionalGeneration.from_pretrained(MODEL_NAME, return_dict=True)

tokenizer = T5Tokenizer.from_pretrained(TOKENIZER_NAME)