Upload 8 files

tagger/tagger.py

@@ -189,7 +189,7 @@ def convert_danbooru_to_e621_prompt(input_prompt: str = "", prompt_type: str = "
 
 from translatepy import Translator
 translator = Translator()
-def translate_prompt(prompt: str = ""):
+def translate_prompt_old(prompt: str = ""):
     def translate_to_english(input: str):
         try:
             output = str(translator.translate(input, 'English'))
@@ -218,6 +218,15 @@ def translate_prompt(prompt: str = ""):
     return ", ".join(outputs)
 
 
+def translate_prompt(input: str):
+    try:
+        output = str(translator.translate(input, 'English'))
+    except Exception as e:
+        output = input
+        print(e)
+    return output
+
+
 def translate_prompt_to_ja(prompt: str = ""):
     def translate_to_japanese(input: str):
         try: