Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 os.system('!python -m pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html')
 os.system('!git clone -b add_dit_inference_bis https://github.com/NielsRogge/unilm.git')
 
-import argparse
 import cv2
 from unilm.dit.object_detection.ditod import add_vit_config
 from detectron2.config import get_cfg
@@ -10,6 +9,8 @@ from detectron2.utils.visualizer import ColorMode, Visualizer
 from detectron2.data import MetadataCatalog
 from detectron2.engine import DefaultPredictor
 
+import gradio as gr
+
 
 # Step 1: instantiate config
 cfg = get_cfg()