Update app.py
Browse files
app.py
CHANGED
|
@@ -12,14 +12,14 @@ def generate_summary(text, style):
|
|
| 12 |
input_text = f"summarize {style}: {text}"
|
| 13 |
input_words = len(text.split())
|
| 14 |
|
| 15 |
-
if
|
| 16 |
max_len = int(input_words * 0.35)
|
| 17 |
min_len = 5
|
| 18 |
rep_penalty = 2.5
|
| 19 |
length_penalty = 1.5
|
| 20 |
beam_size = 4
|
| 21 |
max_cap = 120
|
| 22 |
-
elif
|
| 23 |
max_len = int(input_words * 0.50)
|
| 24 |
min_len = 20
|
| 25 |
rep_penalty = 1.5
|
|
|
|
| 12 |
input_text = f"summarize {style}: {text}"
|
| 13 |
input_words = len(text.split())
|
| 14 |
|
| 15 |
+
if style == 'harsh':
|
| 16 |
max_len = int(input_words * 0.35)
|
| 17 |
min_len = 5
|
| 18 |
rep_penalty = 2.5
|
| 19 |
length_penalty = 1.5
|
| 20 |
beam_size = 4
|
| 21 |
max_cap = 120
|
| 22 |
+
elif style == 'balanced':
|
| 23 |
max_len = int(input_words * 0.50)
|
| 24 |
min_len = 20
|
| 25 |
rep_penalty = 1.5
|