prajdabre commited on
Commit
3a21e36
·
1 Parent(s): bd34b3e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -35,6 +35,8 @@ def generate(input, slang, tlang):
35
  bos_token_id=bos_id, eos_token_id=eos_id, decoder_start_token_id=tokenizer._convert_token_to_id_with_added_voc("<2"+tlang+">"))
36
  decoded_output = tokenizer.decode(
37
  model_output[0], skip_special_tokens=True, clean_up_tokenization_spaces=False)
 
 
38
 
39
 
40
  return decoded_output
 
35
  bos_token_id=bos_id, eos_token_id=eos_id, decoder_start_token_id=tokenizer._convert_token_to_id_with_added_voc("<2"+tlang+">"))
36
  decoded_output = tokenizer.decode(
37
  model_output[0], skip_special_tokens=True, clean_up_tokenization_spaces=False)
38
+ else:
39
+ decoded_output = input
40
 
41
 
42
  return decoded_output