Update app.py

app.py

@@ -41,27 +41,35 @@ model_types = {
     "pp_layout_publaynet": ["text", "title", "list", "table", "figure"],
     "pp_layout_table": ["table"],
     "yolov8n_layout_paper": [
-        "text",
-        "title",
-        "figure",
-        "figure_caption",
-        "table",
-        "table_caption",
-        "header",
-        "footer",
-        "reference",
-        "equation",
+        "Text",
+        "Title",
+        "Header",
+        "Footer",
+        "Figure",
+        "Table",
+        "Toc",
+        "Figure caption",
+        "Table caption",
     ],
     "yolov8n_layout_report": [
-        "text",
-        "title",
-        "header",
-        "footer",
-        "figure",
-        "figure_caption",
-        "table",
-        "table_caption",
-        "toc",
+        "Text",
+        "Title",
+        "Header",
+        "Footer",
+        "Figure",
+        "Table",
+        "Toc",
+        "Figure caption",
+        "Table caption",
+    ],
+    "yolov8n_layout_publaynet": ["Text", "Title", "List", "Table", "Figure"],
+    "yolov8n_layout_general6": [
+        "Text",
+        "Title",
+        "Figure",
+        "Table",
+        "Caption",
+        "Equation",
     ],
 }
 select_model = st.selectbox("选择版面分析模型：", model_types.keys())
@@ -85,3 +93,4 @@ if img_file_buffer:
         ploted_img = VisLayout.draw_detections(img, boxes, scores, class_names)
 
     st.image(ploted_img, use_column_width=True)
+