Update translate.py
Browse files- translate.py +2 -2
translate.py
CHANGED
@@ -67,9 +67,9 @@ def translate(
|
|
67 |
|
68 |
|
69 |
from config import load_config
|
70 |
-
from
|
71 |
if __name__ == '__main__':
|
72 |
-
config = load_config(file_name='
|
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(
|