Update app.py

app.py

@@ -26,7 +26,8 @@ def summarize(text, model, tokenizer, num_beams=4, device='cpu'):
         summary_ids = model.generate(inputs, max_length=256, num_beams=num_beams)
     summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
     return summary
-
+    
+text = st.text_area('Nhập tiêu đề vào đây')
 if text:
     re1 = summarize(model_org, tokenizer, text)
     re2 = summarize(model_aug, tokenizer, text)