File size: 702 Bytes
3bea7ba
 
 
3737c61
 
 
 
c7deb73
3737c61
be692a1
3737c61
 
4cccde5
 
771a3ac
 
dd1408d
 
 
 
4cccde5
 
75c5460
3737c61
 
75c5460
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import subprocess
subprocess.run(["pip", "install", "./textgen.zip"])

import gradio as gr
from textgen import TextGenerator

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

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

with gr.Blocks() as demo:
    gr.Markdown("# 🦊 textgen")
    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")

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

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