VMORnD commited on
Commit
c6358d1
1 Parent(s): 471175f

Update core.py

Browse files
Files changed (1) hide show
  1. core.py +2 -0
core.py CHANGED
@@ -51,6 +51,7 @@ def text_summarize_en(text_input):
51
  outputs = text_summarization_model_En.generate(
52
  input_ids=input_ids, attention_mask=attention_masks,
53
  max_length=256,
 
54
  early_stopping=True
55
  )
56
  text = ""
@@ -65,6 +66,7 @@ def text_summarize_vi(text_input):
65
  outputs = text_summarization_model_Vi.generate(
66
  input_ids=input_ids, attention_mask=attention_masks,
67
  max_length=256,
 
68
  early_stopping=True
69
  )
70
  text = ""
51
  outputs = text_summarization_model_En.generate(
52
  input_ids=input_ids, attention_mask=attention_masks,
53
  max_length=256,
54
+ min_length=20,
55
  early_stopping=True
56
  )
57
  text = ""
66
  outputs = text_summarization_model_Vi.generate(
67
  input_ids=input_ids, attention_mask=attention_masks,
68
  max_length=256,
69
+ min_length=20,
70
  early_stopping=True
71
  )
72
  text = ""