TiberiuCristianLeon commited on
Commit
4bd448b
·
verified ·
1 Parent(s): 1928e5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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], message_text
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]: