Update app.py
Browse files
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)
|