Views
No views yet
1from transformers import T5ForConditionalGeneration, AutoTokenizer
2tokenizer = AutoTokenizer.from_pretrained(
3 'mesolitica/translation-t5-small-standard-bahasa-cased-v2',
4 use_fast=False
5)
6model = T5ForConditionalGeneration.from_pretrained(
7 'mesolitica/translation-t5-small-standard-bahasa-cased-v2'
8)
9s = 'Hai, ada yang bisa saya bantu?'
10input_ids = tokenizer.encode(f'terjemah ke Melayu: {s}', return_tensors = 'pt')
11outputs = model.generate(input_ids, max_length = 100)
12all_special_ids = [0, 1, 2]
13outputs = [i for i in outputs[0] if i not in all_special_ids]
14print(tokenizer.decode(outputs, spaces_between_special_tokens = False))