JaMe76 commited on
Commit
981daf7
1 Parent(s): 37bf817

Update app.py

Browse files

Replace names with LayoutType

Files changed (1) hide show
  1. app.py +5 -5
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.names.C.TEXT,
25
- "2": dd.names.C.TITLE,
26
- "3": dd.names.C.LIST,
27
- "4": dd.names.C.TAB,
28
- "5": dd.names.C.FIG},
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():