Spaces:
Runtime error
Runtime error
Added Translator
Browse files
main.py
CHANGED
@@ -1,19 +1,17 @@
|
|
1 |
-
|
2 |
from lang_list import LANGUAGE_NAME_TO_CODE
|
3 |
import runpod
|
4 |
|
5 |
def run_t2tt(input_text: str, source_language: str, target_language: str) -> str:
|
6 |
source_language_code = LANGUAGE_NAME_TO_CODE[source_language]
|
7 |
target_language_code = LANGUAGE_NAME_TO_CODE[target_language]
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
import json
|
16 |
-
return json.dumps({"input_text": input_text, "src_code": source_language_code, "tgt_code": target_language_code})
|
17 |
|
18 |
def runpod_handler(job):
|
19 |
job_input = job['input']
|
|
|
1 |
+
from translator import translator
|
2 |
from lang_list import LANGUAGE_NAME_TO_CODE
|
3 |
import runpod
|
4 |
|
5 |
def run_t2tt(input_text: str, source_language: str, target_language: str) -> str:
|
6 |
source_language_code = LANGUAGE_NAME_TO_CODE[source_language]
|
7 |
target_language_code = LANGUAGE_NAME_TO_CODE[target_language]
|
8 |
+
out_texts, _ = translator.predict(
|
9 |
+
input=input_text,
|
10 |
+
task_str="T2TT",
|
11 |
+
src_lang=source_language_code,
|
12 |
+
tgt_lang=target_language_code,
|
13 |
+
)
|
14 |
+
return str(out_texts[0])
|
|
|
|
|
15 |
|
16 |
def runpod_handler(job):
|
17 |
job_input = job['input']
|