Test commited on
Commit
6fccb8e
·
verified ·
1 Parent(s): 694327e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ def d(input):
9
  **model_inputs,
10
  forced_bos_token_id=tokenizer.lang_code_to_id["de_DE"]
11
  )
12
- output_DE = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
13
- output_DE = [s.strip("[]' ") for s in output_DE]
14
  return output_DE
15
 
16
  iface = gr.Interface(fn=d, inputs="text", outputs="text")
 
9
  **model_inputs,
10
  forced_bos_token_id=tokenizer.lang_code_to_id["de_DE"]
11
  )
12
+ output_DE = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
13
+ output_DE = output_DE.strip("[]' ")
14
  return output_DE
15
 
16
  iface = gr.Interface(fn=d, inputs="text", outputs="text")