alert warning

app.py

@@ -34,6 +34,7 @@ for i in range(st.session_state.num_docs):
 category = col1.selectbox("Chọn chủ để của văn bản: ", ['Giáo dục', 'Giải trí - Thể thao', 'Khoa học - Công nghệ', 'Kinh tế', 'Pháp luật', 'Thế giới', 'Văn hóa - Xã hội', 'Đời sống'])
 
 def summarize():
+    st.warning("Mô hình ViT5 sẽ mất 2 đến 3 phút để tóm tắt. Xin hãy kiên nhẫn đợi một chút!")
     summ, _ = infer(st.session_state.docs, category)
     with col2.container():
         col2.subheader("Kết quả: ")
@@ -41,7 +42,10 @@ def summarize():
         col2.write("Sử dụng CHDG:")
         col2.write(summ)
         summ_vit5 = vit5_infer(st.session_state.docs)
-        col2.write(summ_vit5)
+        col2.write("Sử dụng ViT5:")
+        for i in range(len(summ_vit5)):
+            col2.write(f"Bản tóm tắt {i}")
+            col2.write(f"{summ_vit5[i]}")
     
 if col1.button("Tóm tắt"):
     summarize()