Spaces:
Runtime error
Runtime error
gorkemozkaya
commited on
Commit
•
8e7bfe1
1
Parent(s):
e542316
fix more path issues
Browse files
main.py
CHANGED
@@ -8,13 +8,13 @@ from official.core import exp_factory, task_factory
|
|
8 |
from official.nlp.configs import wmt_transformer_experiments as wmt_te
|
9 |
|
10 |
tokenizer= tftxt.SentencepieceTokenizer(
|
11 |
-
model=tf.io.gfile.GFile("/
|
12 |
add_eos=True)
|
13 |
|
14 |
# sys.path = ['/root/.local/lib/python3.10/site-packages', '/root/.local/bin'] + sys.path
|
15 |
|
16 |
task_config = exp_factory.get_exp_config('transformer_tr_en_blended/base').task
|
17 |
-
task_config.sentencepiece_model_path = 'pretrained_v2/sentencepiece_en_tr.model'
|
18 |
|
19 |
translation_task = task_factory.get_task(task_config)
|
20 |
model_en_tr = translation_task.build_model()
|
|
|
8 |
from official.nlp.configs import wmt_transformer_experiments as wmt_te
|
9 |
|
10 |
tokenizer= tftxt.SentencepieceTokenizer(
|
11 |
+
model=tf.io.gfile.GFile("/code/pretrained_v2/sentencepiece_en_tr.model", "rb").read(),
|
12 |
add_eos=True)
|
13 |
|
14 |
# sys.path = ['/root/.local/lib/python3.10/site-packages', '/root/.local/bin'] + sys.path
|
15 |
|
16 |
task_config = exp_factory.get_exp_config('transformer_tr_en_blended/base').task
|
17 |
+
task_config.sentencepiece_model_path = '/code/pretrained_v2/sentencepiece_en_tr.model'
|
18 |
|
19 |
translation_task = task_factory.get_task(task_config)
|
20 |
model_en_tr = translation_task.build_model()
|