Wootang01 commited on
Commit
61a1da3
·
1 Parent(s): c649c75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,8 +27,8 @@ translated_text = en_to_de_output[0]['translation_text']
27
 
28
  input_ids = tokenizer_de2en.encode(translated_text, return_tensors="pt")
29
  output_ids = de2en.generate(input_ids)[0]
30
- augmented_text = tokenizer_de2en.decode(output_ids, skip_special_tokens=True)
31
 
32
- st.write("Paraphrased text using back translation: ", augmented_text)
33
 
34
 
 
27
 
28
  input_ids = tokenizer_de2en.encode(translated_text, return_tensors="pt")
29
  output_ids = de2en.generate(input_ids)[0]
30
+ augmented_text = tokenizer_de2en.decode(output_ids, skip_special_tokens=False)
31
 
32
+ st.write("Paraphrased text: ", augmented_text)
33
 
34