Update app.py

app.py

@@ -71,6 +71,18 @@ model_types = {
         "Caption",
         "Equation",
     ],
+    "doclayout_yolo": [
+        "title",
+        "text",
+        "abandon",
+        "figure",
+        "figure_caption",
+        "table",
+        "table_caption",
+        "table_footnote",
+        "isolate_formula",
+        "formula_caption",
+    ],
 }
 select_model = st.selectbox("选择版面分析模型：", model_types.keys())
 st.write("支持检测类型：")
@@ -93,4 +105,3 @@ if img_file_buffer:
         ploted_img = VisLayout.draw_detections(img, boxes, scores, class_names)
 
     st.image(ploted_img, use_column_width=True)
-