Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 10:35:10 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Make src and dst languages explicit

Browse source
This commit is contained in:
oobabooga 2023-01-28 23:29:17 -03:00
parent 7ff68ef252
commit 1a8d815de4
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
extensions/google-translate/script.py
View file

@ -12,11 +12,11 @@ def input_modifier(string):
they are fed into the model.
"""
return translator.translate(string, dest='en').text
return translator.translate(string, src=params['language 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
return translator.translate(string, src="en", dest=params['language string']).text