import gradio as gr
from transformers import pipeline

model = pipeline("text-generation", model="MarkProMaster229/arch")


def generate_text(prompt):
    return model(prompt, max_length=50)[0]['generated_text']

demo = gr.Interface(
    fn=generate_text,
    inputs=gr.Textbox(lines=2, placeholder="Введите текст..."),
    outputs="text"
)

demo.launch()