Update app.py

app.py

@@ -8,10 +8,10 @@ model = AutoModelForSeq2SeqLM.from_pretrained("VietAI/vit5-large-vietnews-summar
 def preprocess(inp):
     text = "summarize: " + inp + " </s>"
     features = tokenizer(text, return_tensors="pt")
-    return features['input_ids'], features['attention_masks']
-def predict(input_ids, attention_masks):
+    return features['input_ids'], features['attention_mask']
+def predict(input_ids, attention_mask):
     outputs = model.generate(
-        input_ids=input_ids, attention_mask=attention_masks,
+        input_ids=input_ids, attention_mask=attention_mask,
         max_length=256,
         early_stopping=True,
     )