Update app.py
Browse files
app.py
CHANGED
|
@@ -1,5 +1,8 @@
|
|
| 1 |
-
import
|
|
|
|
|
|
|
| 2 |
|
|
|
|
| 3 |
import cv2
|
| 4 |
from ditod import add_vit_config
|
| 5 |
from detectron2.config import get_cfg
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
os.system('!python -m pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html')
|
| 3 |
+
os.system('!git clone -b add_dit_inference_bis https://github.com/NielsRogge/unilm.git')
|
| 4 |
|
| 5 |
+
import argparse
|
| 6 |
import cv2
|
| 7 |
from ditod import add_vit_config
|
| 8 |
from detectron2.config import get_cfg
|