Spaces:

nielsr
/

dit-document-layout-analysis

Running

nielsr HF staff commited on Mar 9, 2022

Commit

4b41ae8

•

1 Parent(s): c023813

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ 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 3: set device
 # TODO also support GPU

 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 3: set device
 # TODO also support GPU