Views
No views yet
transformers library installed. You can install it via pip:pip install transformers1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3# Load the tokenizer and model
4model_name = "6yuru99/medical-nllb-200-en2zh_hant"
5model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
6tokenizer = AutoTokenizer.from_pretrained(model_name, src_lang="zho_Hant")
7
8# Example Chinese medical text
9input_text = "患者有高血壓病史,需定期檢查血壓。"
10
11# Tokenize the input text
12inputs = tokenizer(input_text, return_tensors="pt")
13
14# Generate translation
15translated_tokens = model.generate(**inputs, forced_bos_token_id=tokenizer.convert_tokens_to_ids("eng_Latn"))
16
17# Decode the translated text
18outputs = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
19
20print("Translated Text:", outputs)