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 + 5
+            end = start + 6
             break
             
     # Extract abstract 
@@ -32,14 +32,14 @@ def summarize_text(text):
 
     # Tokenize abstract 
     inputs = tokenizer(abstract,
-                       #max_length=512,
+                       max_length=512,
                        return_tensors="pt",
                        truncation=True)
 
     # Generate summary
     summary_ids = model.generate(inputs['input_ids'],
-                                 #num_beams=3,
-                                 max_length=24,
+                                 num_beams=3,
+                                 max_length=30,
                                  min_length=21,
                                  do_sample=False,
                                  early_stopping=True)