Spaces:

nielsr
/

dit-document-layout-analysis

Running

nielsr HF staff commited on Mar 9, 2022

Commit

b972826

1 Parent(s): 6091fb7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 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_cascade.pth"
 # Step 3: set device
 # TODO also support GPU