Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -41,27 +41,35 @@ model_types = {
|
|
41 |
"pp_layout_publaynet": ["text", "title", "list", "table", "figure"],
|
42 |
"pp_layout_table": ["table"],
|
43 |
"yolov8n_layout_paper": [
|
44 |
-
"
|
45 |
-
"
|
46 |
-
"
|
47 |
-
"
|
48 |
-
"
|
49 |
-
"
|
50 |
-
"
|
51 |
-
"
|
52 |
-
"
|
53 |
-
"equation",
|
54 |
],
|
55 |
"yolov8n_layout_report": [
|
56 |
-
"
|
57 |
-
"
|
58 |
-
"
|
59 |
-
"
|
60 |
-
"
|
61 |
-
"
|
62 |
-
"
|
63 |
-
"
|
64 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
65 |
],
|
66 |
}
|
67 |
select_model = st.selectbox("ιζ©ηι’εζ樑εοΌ", model_types.keys())
|
@@ -85,3 +93,4 @@ if img_file_buffer:
|
|
85 |
ploted_img = VisLayout.draw_detections(img, boxes, scores, class_names)
|
86 |
|
87 |
st.image(ploted_img, use_column_width=True)
|
|
|
|
41 |
"pp_layout_publaynet": ["text", "title", "list", "table", "figure"],
|
42 |
"pp_layout_table": ["table"],
|
43 |
"yolov8n_layout_paper": [
|
44 |
+
"Text",
|
45 |
+
"Title",
|
46 |
+
"Header",
|
47 |
+
"Footer",
|
48 |
+
"Figure",
|
49 |
+
"Table",
|
50 |
+
"Toc",
|
51 |
+
"Figure caption",
|
52 |
+
"Table caption",
|
|
|
53 |
],
|
54 |
"yolov8n_layout_report": [
|
55 |
+
"Text",
|
56 |
+
"Title",
|
57 |
+
"Header",
|
58 |
+
"Footer",
|
59 |
+
"Figure",
|
60 |
+
"Table",
|
61 |
+
"Toc",
|
62 |
+
"Figure caption",
|
63 |
+
"Table caption",
|
64 |
+
],
|
65 |
+
"yolov8n_layout_publaynet": ["Text", "Title", "List", "Table", "Figure"],
|
66 |
+
"yolov8n_layout_general6": [
|
67 |
+
"Text",
|
68 |
+
"Title",
|
69 |
+
"Figure",
|
70 |
+
"Table",
|
71 |
+
"Caption",
|
72 |
+
"Equation",
|
73 |
],
|
74 |
}
|
75 |
select_model = st.selectbox("ιζ©ηι’εζ樑εοΌ", model_types.keys())
|
|
|
93 |
ploted_img = VisLayout.draw_detections(img, boxes, scores, class_names)
|
94 |
|
95 |
st.image(ploted_img, use_column_width=True)
|
96 |
+
|