Update app.py

app.py

@@ -71,18 +71,9 @@ model_types = {
         "Caption",
         "Equation",
     ],
-    "doclayout_yolo": [
-        "title",
-        "text",
-        "abandon",
-        "figure",
-        "figure_caption",
-        "table",
-        "table_caption",
-        "table_footnote",
-        "isolate_formula",
-        "formula_caption",
-    ],
+    "doclayout_docstructbench": ['title', 'plain text', 'abandon', 'figure', 'figure_caption', 'table', 'table_caption', 'table_footnote', 'isolate_formula', 'formula_caption'],
+    "doclayout_d4la": ['DocTitle', 'ParaTitle', 'ParaText', 'ListText', 'RegionTitle', 'Date', 'LetterHead', 'LetterDear', 'LetterSign', 'Question', 'OtherText', 'RegionKV', 'RegionList', 'Abstract', 'Author', 'TableName', 'Table', 'Figure', 'FigureName', 'Equation', 'Reference', 'Footer', 'PageHeader', 'PageFooter', 'Number', 'Catalog', 'PageNumber'],
+    "doclayout_docsynth": ['Caption', 'Footnote', 'Formula', 'List-item', 'Page-footer', 'Page-header', 'Picture', 'Section-header', 'Table', 'Text', 'Title']
 }
 select_model = st.selectbox("选择版面分析模型：", model_types.keys())
 st.write("支持检测类型：")