Add library name to model card

README.md

@@ -1,5 +1,6 @@
 ---
-license: mit
+base_model:
+- meta-llama/Meta-Llama-3-8B
 datasets:
 - NiuTrans/ComMT
 language:
@@ -7,12 +8,12 @@ language:
 - zh
 - de
 - cs
+license: mit
 metrics:
 - bleu
 - comet
-base_model:
-- meta-llama/Meta-Llama-3-8B
 pipeline_tag: translation
+library_name: transformers
 ---
 
 # LaMaTE
@@ -48,7 +49,9 @@ tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
 config = AutoConfig.from_pretrained(model_name_or_path, trust_remote_code=True)
 model = LlamaCrossAttentionEncDec.from_pretrained(model_name_or_path, config=config)
 
-prompt = "Translate the following text from English into Chinese.\nEnglish: The harder you work at it, the more progress you will make.\nChinese: ",
+prompt = "Translate the following text from English into Chinese.
+English: The harder you work at it, the more progress you will make.
+Chinese: ",
 input_ids = tokenizer(prompt, return_tensors="pt")
 outputs_tokenized = model.generate(
     **input_ids,
@@ -71,4 +74,4 @@ print(outputs)
       archivePrefix={arXiv},
       primaryClass={cs.CL}
 }
-```
+```