import gradio as gr
import spaces
from transformers import pipeline
import os

pipe = pipeline("summarization", model="ShynBui/Bartpho_spelling_correction")

def greet(text):
    return pipe(text, max_new_tokens= 512)[0]['summary_text']

demo = gr.Interface(fn=greet, inputs=gr.Text(), outputs=gr.Text(), examples=eval(os.environ['DES_EXAMPLE']))
demo.launch()