nielsr HF staff commited on
Commit
90ea4db
1 Parent(s): 5b70501

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ os.system('!git clone -b add_dit_inference_bis https://github.com/NielsRogge/uni
4
 
5
  import argparse
6
  import cv2
7
- from ditod import add_vit_config
8
  from detectron2.config import get_cfg
9
  from detectron2.utils.visualizer import ColorMode, Visualizer
10
  from detectron2.data import MetadataCatalog
 
4
 
5
  import argparse
6
  import cv2
7
+ from unilm.dit.object_detection.ditod import add_vit_config
8
  from detectron2.config import get_cfg
9
  from detectron2.utils.visualizer import ColorMode, Visualizer
10
  from detectron2.data import MetadataCatalog