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)