Add ckp_dir and tokenizer path

[xunuohope1107]

The checkpoint path and tokenizer path should be passed outside, not hard code.