Update app.py

app.py

@@ -28,13 +28,13 @@ def summarize_text(text):
 
     # Load BART model & tokenizer
     tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
-    model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn", low_cpu_mem_usage=True)
+    model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
 
     # Tokenize abstract 
     inputs = tokenizer(abstract, return_tensors="pt", truncation=True)
 
     # Generate summary
-    summary_ids = model.generate(inputs['input_ids'], num_beams=5, max_length=45, min_length=30, early_stopping=True)
+    summary_ids = model.generate(inputs['input_ids'], num_beams=4, max_length=35, min_length=30, early_stopping=True)
     summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
 
     return summary