lityops commited on
Commit
42021b5
·
verified ·
1 Parent(s): ed4fa48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 style_query == '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_query == 'balanced':
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