소스 검색

Add Google Translate extension

oobabooga 3 년 전
부모
커밋
7ff68ef252
2개의 변경된 파일23개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      extensions/google-translate/requirements.txt
  2. 22 0
      extensions/google-translate/script.py

+ 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