Update app.py

app.py

@@ -5,12 +5,14 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 model_id = "hramphul/bart-large-cnn"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
+model.eval()
 
 def summarize(text, max_len, min_len):
     inputs = tokenizer(text, return_tensors="pt", max_length=1024, truncation=True)
     with torch.no_grad():
         ids = model.generate(
             inputs["input_ids"],
+            attention_mask=inputs["attention_mask"],
             max_length=int(max_len),
             min_length=int(min_len),
             num_beams=4,