Update README.md

README.md

@@ -38,7 +38,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 # Load the tokenizer and model
 model_name = "6yuru99/medical-nllb-200-en2zh_hant"
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name, src_lang="eng_Latn")
+tokenizer = AutoTokenizer.from_pretrained(model_name, src_lang="zho_Hant")
 
 # Example Chinese medical text
 input_text = "患者有高血壓病史，需定期檢查血壓。"
@@ -47,7 +47,7 @@ input_text = "患者有高血壓病史，需定期檢查血壓。"
 inputs = tokenizer(input_text, return_tensors="pt")
 
 # Generate translation
-translated_tokens = model.generate(**inputs, forced_bos_token_id=tokenizer.convert_tokens_to_ids("zho_Hant"))
+translated_tokens = model.generate(**inputs, forced_bos_token_id=tokenizer.convert_tokens_to_ids("eng_Latn"))
 
 # Decode the translated text
 outputs = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]