Test
commited on
Update app.py
Browse files
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 =
|
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")
|