Spaces:
Sleeping
Sleeping
no message
Browse files
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=
|
111 |
no_repeat_ngram_size=3,
|
112 |
-
length_penalty=
|
113 |
min_length=150,
|
114 |
-
max_length=
|
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 |
|