moiduy04 commited on
Commit
5d0c765
1 Parent(s): 95e0aba

Update translate.py

Browse files
Files changed (1) hide show
  1. translate.py +2 -2
translate.py CHANGED
@@ -67,9 +67,9 @@ def translate(
67
 
68
 
69
  from config import load_config
70
- from load_and_save_model import load_model_tokenizer
71
  if __name__ == '__main__':
72
- config = load_config(file_name='config_huge.yaml')
73
  model, src_tokenizer, tgt_tokenizer = load_model_tokenizer(config)
74
  text = "ສະບາຍດີ" # Hello.
75
  translation, attn = translate(
 
67
 
68
 
69
  from config import load_config
70
+ from load_model import load_model_tokenizer
71
  if __name__ == '__main__':
72
+ config = load_config(file_name='/config/config_final.yaml')
73
  model, src_tokenizer, tgt_tokenizer = load_model_tokenizer(config)
74
  text = "ສະບາຍດີ" # Hello.
75
  translation, attn = translate(