host / app.py
MarkProMaster229's picture
Update app.py
800a0d5 verified
raw
history blame contribute delete
367 Bytes
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()