Просмотр исходного кода

Add Google Translate extension

oobabooga 3 лет назад
Родитель
Сommit
7ff68ef252

+ 1 - 0
extensions/google-translate/requirements.txt

@@ -0,0 +1 @@
+googletrans==3.1.0a0

+ 22 - 0
extensions/google-translate/script.py

@@ -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