import streamlit as st | |
from transformers import pipeline | |
# from pathlib import Path | |
st.title("Thai News Summarizer") | |
st.subheader("Thai News Summarizer") | |
text = st.text_area('News', '', height=200) | |
# st.write('This KHAW? -->', text) | |
if st.button('Summarize'): | |
summarizer = pipeline("summarization", model = "Saltywan/mt5-base-thaisum-news" ) | |
text = summarizer("summarize: " + text, max_length=256) | |
desired_text = text[0]['summary_text'] | |
st.write(desired_text) |