PhongLT commited on
Commit
e0a44f2
·
1 Parent(s): 1617d73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -13,16 +13,16 @@ model = AutoModelForSeq2SeqLM.from_pretrained("PhongLT/ViLexNorm-bartpho-syllabl
13
  def normalize(input):
14
  input_ids = tokenizer( input,
15
  return_tensors="pt",
16
- max_length=512,
17
  padding="max_length",
18
  truncation= True).input_ids
19
 
20
  output_ids = model.generate(input_ids,
21
- max_length=512)
22
 
23
  return tokenizer.decode(output_ids[0],
24
  skip_special_tokens=True,
25
- max_length=512)
26
 
27
  # Create title, description and article strings
28
  title = "Vietnamese Lexical Normalization"
 
13
  def normalize(input):
14
  input_ids = tokenizer( input,
15
  return_tensors="pt",
16
+ max_length=128,
17
  padding="max_length",
18
  truncation= True).input_ids
19
 
20
  output_ids = model.generate(input_ids,
21
+ max_length=128)
22
 
23
  return tokenizer.decode(output_ids[0],
24
  skip_special_tokens=True,
25
+ max_length=128)
26
 
27
  # Create title, description and article strings
28
  title = "Vietnamese Lexical Normalization"