Spaces:
Runtime error
Runtime error
Update app.py
Browse filesReplace names with LayoutType
app.py
CHANGED
@@ -21,11 +21,11 @@ dd.ModelCatalog.register("layout/model_final_inf_only.pt",ModelProfile(
|
|
21 |
hf_repo_id=os.environ.get("HF_REPO"),
|
22 |
hf_model_name="model_final_inf_only.pt",
|
23 |
hf_config_file=["Base-RCNN-FPN.yaml", "CASCADE_RCNN_R_50_FPN_GN.yaml"],
|
24 |
-
categories={"1": dd.
|
25 |
-
"2": dd.
|
26 |
-
"3": dd.
|
27 |
-
"4": dd.
|
28 |
-
"5": dd.
|
29 |
))
|
30 |
|
31 |
def get_space_dd_analyzer():
|
|
|
21 |
hf_repo_id=os.environ.get("HF_REPO"),
|
22 |
hf_model_name="model_final_inf_only.pt",
|
23 |
hf_config_file=["Base-RCNN-FPN.yaml", "CASCADE_RCNN_R_50_FPN_GN.yaml"],
|
24 |
+
categories={"1": dd.LayoutType.text,
|
25 |
+
"2": dd.LayoutType.title,
|
26 |
+
"3": dd.LayoutType.list,
|
27 |
+
"4": dd.LayoutType.table,
|
28 |
+
"5": dd.LayoutType.figure},
|
29 |
))
|
30 |
|
31 |
def get_space_dd_analyzer():
|