app.py

import gradio as gr
from transformers import pipeline

pipe = pipeline("text2text-generation", model="yxccai/text-style")

def chat(input_text):
    result = pipe(input_text, max_new_tokens=512)[0]["generated_text"]
    return result

gr.Interface(fn=chat, inputs="text", outputs="text").launch()