Spaces:

nielsr
/

dit-document-layout-analysis

Running

nielsr HF staff commited on Mar 9, 2022

Commit

b06234a

•

1 Parent(s): a07666b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ def add_vit_config(cfg):
 # Step 1: instantiate config
 cfg = get_cfg()
 add_vit_config(cfg)
-cfg.merge_from_file("./cascade_dit_base.yaml")
 # Step 2: add model weights URL to config
 cfg.MODEL.WEIGHTS = "https://layoutlm.blob.core.windows.net/dit/dit-fts/publaynet_dit-b_mrcnn.pth"

 # Step 1: instantiate config
 cfg = get_cfg()
 add_vit_config(cfg)
+cfg.merge_from_file("cascade_dit_base.yml")
 # Step 2: add model weights URL to config
 cfg.MODEL.WEIGHTS = "https://layoutlm.blob.core.windows.net/dit/dit-fts/publaynet_dit-b_mrcnn.pth"