mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-11-22 08:07:56 +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