👍️[Update] Models : change print log to json format log

2024-08-11 13:51:48 +09:00
parent 3924b2eb3b
commit 633cdf246c
4 changed files with 10 additions and 7 deletions
--- a/src-python/models/translation/translation_translator.py
+++ b/src-python/models/translation/translation_translator.py
@@ -42,7 +42,8 @@ class Translator():
        try:
            self.ctranslate2_tokenizer = transformers.AutoTokenizer.from_pretrained(tokenizer, cache_dir=tokenizer_path)
        except Exception as e:
-            print("Error: changeCTranslate2Model()", e)
+            import json
+            print(json.dumps({"status":348, "log": f"error:changeCTranslate2Model() {e}"}), flush=True)
            tokenizer_path = os.path.join("./weights", "ctranslate2", directory_name, "tokenizer")
            self.ctranslate2_tokenizer = transformers.AutoTokenizer.from_pretrained(tokenizer, cache_dir=tokenizer_path)
        self.is_loaded_ctranslate2_model = True