Marroco93 commited on
Commit
ecb4879
1 Parent(s): 506bda4

no message

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -107,11 +107,11 @@ def summarize_legal_text(text):
107
  # Generate summary
108
  summary_ids = model.generate(
109
  inputs,
110
- num_beams=9,
111
  no_repeat_ngram_size=3,
112
- length_penalty=2.0,
113
  min_length=150,
114
- max_length=250,
115
  early_stopping=True
116
  )
117
 
 
107
  # Generate summary
108
  summary_ids = model.generate(
109
  inputs,
110
+ num_beams=5,
111
  no_repeat_ngram_size=3,
112
+ length_penalty=1.0,
113
  min_length=150,
114
+ max_length=1000,
115
  early_stopping=True
116
  )
117