File size: 367 Bytes
016fcf3
 
 
 
 
 
 
800a0d5
016fcf3
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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()