QINGCHE commited on
Commit
caa3d6b
1 Parent(s): 497b9c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +46 -46
app.py CHANGED
@@ -1,46 +1,46 @@
1
- import numpy as np
2
- import gradio as gr
3
- import textInput
4
- from BERT_inference import BertClassificationModel
5
-
6
-
7
- output = []
8
- keys = []
9
-
10
- # css = ".output {min-height: 500px}"
11
-
12
-
13
- with gr.Blocks(css = ".output {min-height: 500px}") as demo:
14
- #用markdown语法编辑输出一段话
15
- gr.Markdown("# 文本分类系统")
16
- gr.Markdown("请选择要输入的文件或填入文本")
17
- topic_num = gr.Number(label="主题个数")
18
- max_length = gr.Number(label="摘要最大长度")
19
- with gr.Tabs():
20
- with gr.Tab("文本输入"):
21
- text_input = gr.TextArea(lines=10)
22
- text_button = gr.Button("生成")
23
-
24
- with gr.Tab("文件输入"):
25
- gr.Markdown("目前支持的格式有PDF、Word、txt")
26
- file_input = gr.File(file_types=["text", ".pdf", ".docx"])
27
- file_button = gr.Button("生成")
28
- # 设置tab选项卡
29
- with gr.Tabs():
30
- with gr.Tab("分类页"):
31
- text_keys_output = gr.TextArea(lines=30)
32
-
33
- with gr.Tab("摘要页",):
34
- #Blocks特有组件,设置所有子组件按水平排列
35
- text_ab_output = gr.TextArea(lines=30)
36
-
37
- with gr.Tab("下载页"):
38
- file_txt_output = gr.File(label="txt格式")
39
- file_docx_output = gr.File(label="docx格式")
40
- file_pdf_output = gr.File(label="pdf格式")
41
- # with gr.Accordion("Open for More!"):
42
- # gr.Markdown("Look at me...")
43
- text_button.click(textInput.text_dump_to_lines, inputs=[text_input,topic_num,max_length], outputs=[text_keys_output,text_ab_output,file_txt_output,file_docx_output,file_pdf_output])
44
- file_button.click(textInput.file_dump_to_lines,inputs=[file_input,topic_num,max_length], outputs=[text_keys_output,text_ab_output,file_txt_output,file_docx_output,file_pdf_output])
45
-
46
- demo.launch()
 
1
+ import numpy as np
2
+ import gradio as gr
3
+ import textInput
4
+ from BERT_inference import BertClassificationModel
5
+
6
+
7
+ output = []
8
+ keys = []
9
+
10
+ # css = ".output {min-height: 500px}"
11
+
12
+
13
+ with gr.Blocks(css = ".output {min-height: 500px}") as demo:
14
+ #用markdown语法编辑输出一段话
15
+ gr.Markdown("# 文本分类系统")
16
+ gr.Markdown("请选择要输入的文件或填入文本")
17
+ topic_num = gr.Number(label="主题个数")
18
+ max_length = gr.Number(label="摘要最大长度")
19
+ with gr.Tabs():
20
+ with gr.Tab("文本输入"):
21
+ text_input = gr.TextArea(lines=10)
22
+ text_button = gr.Button("生成")
23
+
24
+ with gr.Tab("文件输入"):
25
+ gr.Markdown("目前支持的格式有PDF、Word、txt")
26
+ file_input = gr.File(file_types=["text", ".pdf", ".docx"])
27
+ file_button = gr.Button("生成")
28
+ # 设置tab选项卡
29
+ with gr.Tabs():
30
+ with gr.Tab("分类页"):
31
+ text_keys_output = gr.TextArea(lines=30)
32
+
33
+ with gr.Tab("摘要页",):
34
+ #Blocks特有组件,设置所有子组件按水平排列
35
+ text_ab_output = gr.TextArea(lines=30)
36
+
37
+ with gr.Tab("下载页"):
38
+ file_txt_output = gr.File(label="txt格式")
39
+ file_docx_output = gr.File(label="docx格式")
40
+ file_pdf_output = gr.File(label="pdf格式")
41
+ # with gr.Accordion("Open for More!"):
42
+ # gr.Markdown("Look at me...")
43
+ text_button.click(textInput.text_dump_to_lines, inputs=[text_input,topic_num,max_length], outputs=[text_keys_output,text_ab_output,file_txt_output,file_docx_output,file_pdf_output])
44
+ file_button.click(textInput.file_dump_to_lines,inputs=[file_input,topic_num,max_length], outputs=[text_keys_output,text_ab_output,file_txt_output,file_docx_output,file_pdf_output])
45
+
46
+ demo.launch(show_api=False, enable_queue=False)