Update app.py

app.py

@@ -12,29 +12,36 @@ def generate_summary(text, style):
   input_text = f"summarize {style}: {text}"
   input_words = len(text.split())
 
-  if style == 'harsh':
+  if style_query == 'harsh':
     max_len = int(input_words * 0.35)
     min_len = 5
     rep_penalty = 2.5
+    length_penalty = 1.5
     beam_size = 4
-  elif style == 'standard':
+    max_cap = 120
+  elif style_query == 'balanced':
     max_len = int(input_words * 0.50)
     min_len = 20
     rep_penalty = 1.5
+    length_penalty = 1.2
     beam_size = 4
+    max_cap = 180
   else:
     max_len = int(input_words * 0.70)
     min_len = 50
     rep_penalty = 1.2
+    length_penalty = 0.8
     beam_size = 4
+    max_cap = 256
 
-  max_len = min(max_len, 256)
+  max_len = min(max_len, max_cap)
   
   output = summarizer(
     input_text,
     max_length=max_len, 
     min_length=min_len, 
     num_beams=beam_size,
+    length_penalty=length_penalty,
     repetition_penalty=rep_penalty,
     no_repeat_ngram_size=3,
     early_stopping=True