app.py

import subprocess
subprocess.run(["pip", "install", "./textgen.zip"])

import gradio as gr
from textgen import TextGenerator

textgen = TextGenerator()
textgen.load("textgen.bin")

def generate_examples():
    return [textgen.generate(length=20).splitlines()[0] for _ in range(20)]

def generate(text, length=50):
    return textgen.generate(text, length=length)

with gr.Blocks() as demo:
    gr.Markdown("# 🦊 textgen")
    with gr.Row():
        with gr.Column():
            with gr.Group():
                text = gr.TextArea(label="Text")
            length = gr.Number(label="Length", value=25)
            with gr.Row():
                btn = gr.Button("Generate", variant="primary")
                clear = gr.Button("Clear")
        with gr.Column():
            examples = gr.Examples(generate_examples(), [text])

    btn.click(generate, [text, length], [text])
    clear.click(lambda: "", [], [text])

if __name__ == "__main__":
    demo.queue().launch(debug=True)