Omnibus commited on
Commit
ac6d7f4
Β·
1 Parent(s): 2b5dfaf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -149,7 +149,7 @@ def trans_page(input,trg):
149
  with torch.no_grad():
150
  encoded_input = tokenizer(langs.name, return_tensors="pt").to(device)
151
  generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
152
- trans_langs = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
153
  new_lang.append(trans_langs)
154
  else:
155
  new_lang=lang_id
 
149
  with torch.no_grad():
150
  encoded_input = tokenizer(langs.name, return_tensors="pt").to(device)
151
  generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
152
+ trans_langs = tokenizer.batch_decode(generated_tokens, skip_special_tokens=False)[0]
153
  new_lang.append(trans_langs)
154
  else:
155
  new_lang=lang_id