github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

blackified

Browse Source
This commit is contained in:
Lincoln Stein
2023-07-29 21:14:18 -04:00
parent e73c12cac2
commit 0c31eaee61
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/app/invocations/translate.py
View File

@@ -12,14 +12,16 @@ from .params import StringOutput
translate_available = False
try:
import translators as ts
translate_available = True
except:
pass
DEFAULT_PROMPT = "" if translate_available else "To use this node, please 'pip install --upgrade translators'"
class TranslateInvocation(BaseInvocation):
""" Use the translators package to translate 330 languages into English prompts """
"""Use the translators package to translate 330 languages into English prompts"""
# fmt: off
type: Literal["translate"] = "translate"
@@ -35,6 +37,4 @@ class TranslateInvocation(BaseInvocation):
}
def invoke(self, context: InvocationContext) -> StringOutput:
return StringOutput(
text = ts.translate_text(self.prompt)
)
return StringOutput(text=ts.translate_text(self.prompt))