Views
No views yet
1from hok2han_model import Seq2SeqTransformer
2model = Seq2SeqTransformer.from_pretrained("KikKoh/Hok2Han")
3model.eval()
4
5from transformers import Wav2Vec2Processor, BertTokenizer
6input_processor = Wav2Vec2Processor.from_pretrained("你的輸入tokenizer路徑或repo")
7output_tokenizer = BertTokenizer.from_pretrained("你的輸出tokenizer路徑或repo")
8
9# 範例推論
10output = model(src=input_ids, tgt=tgt_ids,
11 src_pad_idx=input_processor.tokenizer.pad_token_id,
12 tgt_pad_idx=output_tokenizer.pad_token_id)
13
14pred_ids = output.argmax(dim=-1)
15pred_text = output_tokenizer.decode(pred_ids[0], skip_special_tokens=True)
16print(pred_text)