Update README.md

README.md

@@ -19,10 +19,10 @@ Training scripts from [LalitaDeelert/NLP-ZH_TH-Project](https://github.com/Lalit
 
 ```
 export WANDB_PROJECT=marianmt-zh_cn-th
-python train_model.py --input_fname ../data/v1/Train.csv \\\\\\\\
-\\\\\\\\\\\\\\\\t--output_dir ../models/marianmt-zh_cn-th \\\\\\\\
-\\\\\\\\\\\\\\\\t--source_lang zh --target_lang th \\\\\\\\
-\\\\\\\\\\\\\\\\t--metric_tokenize th_syllable --fp16 
+python train_model.py --input_fname ../data/v1/Train.csv \\\\\\\\\\\\\\\\
+\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\t--output_dir ../models/marianmt-zh_cn-th \\\\\\\\\\\\\\\\
+\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\t--source_lang zh --target_lang th \\\\\\\\\\\\\\\\
+\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\t--metric_tokenize th_syllable --fp16 
 ```
 
 ## Usage
@@ -30,8 +30,8 @@ python train_model.py --input_fname ../data/v1/Train.csv \\\\\\\\
 ```
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
  
-tokenizer = AutoTokenizer.from_pretrained("cstorm125/marianmt-zh_cn-th")
-model = AutoModelForSeq2SeqLM.from_pretrained("cstorm125/marianmt-zh_cn-th").cpu()
+tokenizer = AutoTokenizer.from_pretrained("Lalita/marianmt-zh_cn-th")
+model = AutoModelForSeq2SeqLM.from_pretrained("Lalita/marianmt-zh_cn-th").cpu()
 
 src_text = [
     '我爱你',