Add Google Translate extension

This commit is contained in:
oobabooga 2023-01-28 23:26:07 -03:00
parent e349b52256
commit 7ff68ef252
2 changed files with 23 additions and 0 deletions

View file

@ -0,0 +1 @@
googletrans==3.1.0a0

View file

@ -0,0 +1,22 @@
from googletrans import Translator
translator = Translator()
params = {
"language string": "ja",
}
def input_modifier(string):
"""
This function is applied to your text inputs before
they are fed into the model.
"""
return translator.translate(string, dest='en').text
def output_modifier(string):
"""
This function is applied to the model outputs.
"""
return translator.translate(string, dest=params['language string']).text