cutiee82 commited on
Commit
4cccde5
·
1 Parent(s): 1f961fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -18
app.py CHANGED
@@ -7,27 +7,25 @@ from textgen import TextGenerator
7
  textgen = TextGenerator()
8
  textgen.load("textgen.bin")
9
 
10
- def generate_examples():
11
- return gr.Dataset(samples=textgen.generate(length=100).splitlines())
12
-
13
  def generate(text, length=50):
14
  return textgen.generate(text, length=length)
15
 
16
- def main():
17
- with gr.Blocks(title="🦊 textgen") as demo:
18
- gr.Markdown("# 🦊 textgen")
19
- with gr.Group():
20
- text = gr.TextArea(label="Text")
21
- length = gr.Number(label="Length", value=50)
22
- with gr.Row():
 
23
  btn = gr.Button("Generate", variant="primary")
24
- clear_btn = gr.Button("Clear")
25
- examples = gr.Examples([], inputs=text)
26
- demo.load(generate_examples, None, examples.dataset)
27
- btn.click(generate, [text, length], [text])
28
- clear_btn.click(lambda: gr.update(value=""), None, [text])
29
- demo.queue().launch(debug=True)
 
30
 
31
  if __name__ == "__main__":
32
- # Launch interface
33
- main()
 
7
  textgen = TextGenerator()
8
  textgen.load("textgen.bin")
9
 
 
 
 
10
  def generate(text, length=50):
11
  return textgen.generate(text, length=length)
12
 
13
+ with gr.Blocks() as demo:
14
+ gr.Markdown("# 🦊 textgen")
15
+ with gr.Row():
16
+ with gr.Column():
17
+ with gr.Group():
18
+ text = gr.TextArea(label="Text")
19
+ length = gr.Number(label="Length", value=25)
20
+ with gr.Column():
21
  btn = gr.Button("Generate", variant="primary")
22
+ clear = gr.Button("Clear")
23
+ with gr.Row():
24
+ examples = gr.Examples([], [text])
25
+
26
+ demo.load(lambda: textgen.generate(length=100).splitlines(), [], examples)
27
+ btn.click(generate, [text, length], [text])
28
+ clear.click(lambda: "", [], [text])
29
 
30
  if __name__ == "__main__":
31
+ demo.queue().launch(debug=True)