Moleys commited on
Commit
6e66e7c
·
verified ·
1 Parent(s): c32cc47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def translate_text(input_text):
27
  output_tokens = model.generate(**inputs, max_length=512)
28
 
29
  # Giải mã kết quả và viết hoa chữ đầu
30
- translated_text = tokenizer.decode(output_tokens[0], skip_special_tokens=True).title()
31
  translated_lines.append(translated_text)
32
 
33
  return '\n'.join(translated_lines)
 
27
  output_tokens = model.generate(**inputs, max_length=512)
28
 
29
  # Giải mã kết quả và viết hoa chữ đầu
30
+ translated_text = tokenizer.decode(output_tokens[0], skip_special_tokens=True).capwords()
31
  translated_lines.append(translated_text)
32
 
33
  return '\n'.join(translated_lines)