text-generation-webui/extensions/google_translate/script.py

32 lines
716 B
Python
Raw Normal View History

2023-01-29 03:26:07 +01:00
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.
"""
2023-01-29 03:29:17 +01:00
return translator.translate(string, src=params['language string'], dest='en').text
2023-01-29 03:26:07 +01:00
def output_modifier(string):
"""
This function is applied to the model outputs.
"""
2023-01-29 03:29:17 +01:00
return translator.translate(string, src="en", dest=params['language string']).text
def bot_prefix_modifier(string):
"""
This function is only applied in chat mode. It modifies
the prefix text for the Bot and can be used to bias its
behavior.
"""
return string