Spaces:
Sleeping
Sleeping
vteam27
commited on
Commit
·
80ae4f4
1
Parent(s):
af923d2
bug fix
Browse files
app.py
CHANGED
@@ -20,8 +20,8 @@ def run_t2tt(input_text: str, source_language: str, target_language: str) -> str
|
|
20 |
source_language_code = LANGUAGE_NAME_TO_CODE[source_language]
|
21 |
target_language_code = LANGUAGE_NAME_TO_CODE[target_language]
|
22 |
text_inputs = processor(text = input_text, src_lang=source_language_code , return_tensors="pt")
|
23 |
-
|
24 |
-
|
25 |
return str(output)
|
26 |
|
27 |
|
|
|
20 |
source_language_code = LANGUAGE_NAME_TO_CODE[source_language]
|
21 |
target_language_code = LANGUAGE_NAME_TO_CODE[target_language]
|
22 |
text_inputs = processor(text = input_text, src_lang=source_language_code , return_tensors="pt")
|
23 |
+
output_tokens = model.generate(**text_inputs, tgt_lang=target_language_code)
|
24 |
+
output = processor.decode(output_tokens[0].tolist(), skip_special_tokens=True)
|
25 |
return str(output)
|
26 |
|
27 |
|