File size: 374 Bytes
ebbf7e6
a6d6b63
ebbf7e6
a6d6b63
ebbf7e6
 
a6d6b63
 
ebbf7e6
a6d6b63
 
 
 
ebbf7e6
 
a6d6b63
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/finalModel")


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

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

demo.launch()