Update app.py

app.py

@@ -20,7 +20,7 @@ def summarize_text(text):
     for i, sentence in enumerate(sentences):
         if "Abstract" in sentence:
             start = i + 1
-            end = start + 7
+            end = start + 6
             break
             
     # Extract abstract 
@@ -31,10 +31,18 @@ def summarize_text(text):
     model = AutoModelForSeq2SeqLM.from_pretrained("google/bigbird-pegasus-large-arxiv")
 
     # Tokenize abstract 
-    inputs = tokenizer(abstract, max_length=1024, return_tensors="pt", truncation=True)
+    inputs = tokenizer(abstract,
+                       max_length=1024,
+                       return_tensors="pt",
+                       truncation=True)
 
     # Generate summary
-    summary_ids = model.generate(inputs['input_ids'], num_beams=2, max_length=42, min_length=22, do_sample=True, early_stopping=True)
+    summary_ids = model.generate(inputs['input_ids'],
+                                 num_beams=3,
+                                 max_length=40,
+                                 min_length=25,
+                                 do_sample=False,
+                                 early_stopping=True)
     summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
 
     return summary