Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -137,7 +137,7 @@ def mbart(model_name, sl, tl, input_text):
|
|
| 137 |
**encoded,
|
| 138 |
forced_bos_token_id=tokenizer.lang_code_to_id[languagecodes.mbart_large_languages[tl]]
|
| 139 |
)
|
| 140 |
-
return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
|
| 141 |
|
| 142 |
@spaces.GPU
|
| 143 |
def translate_text(input_text: str, sselected_language: str, tselected_language: str, model_name: str) -> tuple[str, str]:
|
|
|
|
| 137 |
**encoded,
|
| 138 |
forced_bos_token_id=tokenizer.lang_code_to_id[languagecodes.mbart_large_languages[tl]]
|
| 139 |
)
|
| 140 |
+
return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
|
| 141 |
|
| 142 |
@spaces.GPU
|
| 143 |
def translate_text(input_text: str, sselected_language: str, tselected_language: str, model_name: str) -> tuple[str, str]:
|