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