Views
No views yet
1import ctranslate2
2import pyonmttok
3import sys
4
5
6if len(sys.argv) < 2:
7 sentence = "I sincerely apologize for not providing the best taste and quality."
8else:
9 sentence = sys.argv[1]
10
11
12tokenizer = pyonmttok.Tokenizer("conservative", joiner_annotate=True)
13tokens = tokenizer(sentence)
14
15model = "/home/techops/data/nmt_data/clean_data_files_v1/ctranslate2/model_4m"
16# model = "/home/techops/data/nmt_data/ctranslate_model/en_ko/100m_300000"
17translator = ctranslate2.Translator(model_path=model, device="cpu")
18outputs = translator.translate_batch([tokens], beam_size=5, num_hypotheses=2, sampling_temperature=0.8, replace_unknowns=True)
19
20translated = outputs[0].hypotheses[0]
21t_s = tokenizer.detokenize(translated)
22
23print(t_s.replace("@@", ""))
24
25>>>
26
27(nmt) [techops@inf-ai-nmt-a01 (screen: ) /data/NMT/2b_nmt/ctranslate]$ python ctran_translate.py
28최고의 맛과 품질을 제공하지 못한 점에 대해 진심으로 사과드립니다.