777

app.py

@@ -1,49 +1,10 @@
-import numpy as np
 import gradio as gr
-import textInput
-from BERT_inference import BertClassificationModel
+import time
 
+def to_black(text):
+    time.sleep(2)
+    output = "hahah"
+    return output
 
-output = []
-keys = []
-
-# css = ".output {min-height: 500px}"
-
-
-with gr.Blocks(css = ".output {min-height: 500px}") as demo:
-    #用markdown语法编辑输出一段话
-    gr.Markdown("# 文本分类系统")
-    gr.Markdown("请选择要输入的文件或填入文本")
-    topic_num = gr.Number(label="主题个数")
-    max_length = gr.Number(label="摘要最大长度")
-    with gr.Tabs():
-        with gr.Tab("文本输入"):
-            text_input = gr.TextArea(lines=10)
-            text_button = gr.Button("生成")
-    
-        with gr.Tab("文件输入"):
-            gr.Markdown("目前支持的格式有PDF、Word、txt")
-            file_input = gr.File(file_types=["text", ".pdf", ".docx"])
-            file_button = gr.Button("生成")
-    # 设置tab选项卡
-    with gr.Tabs():
-        with gr.Tab("分类页"):
-            text_keys_output = gr.TextArea(lines=30)
-
-        with gr.Tab("摘要页",):
-            #Blocks特有组件，设置所有子组件按水平排列
-            text_ab_output = gr.TextArea(lines=30)
-
-        with gr.Tab("下载页"):
-            file_txt_output = gr.File(label="txt格式")
-            file_docx_output = gr.File(label="docx格式")
-            file_pdf_output = gr.File(label="pdf格式")
-    # with gr.Accordion("Open for More!"):
-    #     gr.Markdown("Look at me...")
-    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])
-    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])
-
-try:
-    demo.queue().launch()
-except Exception as e:
-    print("error",e)
+interface = gr.Interface(fn=to_black, inputs="text", outputs="text")
+interface.queue().launch()