mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-11-22 16:17:57 +01:00
Add fixed Google Translation extension
This commit is contained in:
parent
b19415849c
commit
ad148571f4
1
extensions/google_translate/requirements.txt
Normal file
1
extensions/google_translate/requirements.txt
Normal file
@ -0,0 +1 @@
|
|||||||
|
deep-translator==1.9.2
|
29
extensions/google_translate/script.py
Normal file
29
extensions/google_translate/script.py
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
from deep_translator import GoogleTranslator
|
||||||
|
|
||||||
|
params = {
|
||||||
|
"language string": "ja",
|
||||||
|
}
|
||||||
|
|
||||||
|
def input_modifier(string):
|
||||||
|
"""
|
||||||
|
This function is applied to your text inputs before
|
||||||
|
they are fed into the model.
|
||||||
|
"""
|
||||||
|
|
||||||
|
return GoogleTranslator(source=params['language string'], target='en').translate(string)
|
||||||
|
|
||||||
|
def output_modifier(string):
|
||||||
|
"""
|
||||||
|
This function is applied to the model outputs.
|
||||||
|
"""
|
||||||
|
|
||||||
|
return GoogleTranslator(source='en', target=params['language string']).translate(string)
|
||||||
|
|
||||||
|
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
|
Loading…
Reference in New Issue
Block a user