# from transformers import pipeline
import gradio as gr
from pythainlp.tokenize import subword_tokenize


# model = pipeline(
#     "summarization",
# )

def predict(prompt):
    # summary = model(prompt)[0]["summary_text"]
    summary = subword_tokenize(prompt, engine='tcc')
    return summary


# create an interface for the model
with gr.Interface(predict, "textbox", "text") as interface:
    interface.launch()