Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 08:07:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

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