mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-11-21 23:57:58 +01:00
Add Google Translate extension
This commit is contained in:
parent
e349b52256
commit
7ff68ef252
1
extensions/google-translate/requirements.txt
Normal file
1
extensions/google-translate/requirements.txt
Normal file
@ -0,0 +1 @@
|
||||
googletrans==3.1.0a0
|
22
extensions/google-translate/script.py
Normal file
22
extensions/google-translate/script.py
Normal 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
|
Loading…
Reference in New Issue
Block a user