Spaces:

Tzktz
/

Dit-document-layout-analysis

Sleeping

Tzktz commited on Jan 22, 2024

Commit

1dc34dc

verified ·

1 Parent(s): b816ac7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,7 @@
 from unilm.dit.object_detection.ditod import add_vit_config
 import torch
 from detectron2.config import get_cfg
 from detectron2.utils.visualizer import ColorMode, Visualizer
 from detectron2.data import MetadataCatalog

 from unilm.dit.object_detection.ditod import add_vit_config
 import torch
+imort cv2
+from detectron2.config import CfgNode as CN
 from detectron2.config import get_cfg
 from detectron2.utils.visualizer import ColorMode, Visualizer
 from detectron2.data import MetadataCatalog