Fix dit loading

app.py

@@ -8,8 +8,6 @@ sys.path.append("unilm")
 
 import cv2
 
-from unilm.dit.object_detection.ditod import add_vit_config
-
 import torch
 
 from detectron2.config import CfgNode as CN
@@ -18,6 +16,8 @@ from detectron2.utils.visualizer import ColorMode, Visualizer
 from detectron2.data import MetadataCatalog
 from detectron2.engine import DefaultPredictor
 
+from unilm.dit.object_detection.ditod import add_vit_config
+
 import gradio as gr