Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,9 +4,11 @@ import cv2
|
|
| 4 |
import json
|
| 5 |
import os
|
| 6 |
import gradio as gr
|
| 7 |
-
from detectron2.engine import DefaultPredictor
|
| 8 |
-
from detectron2.config import get_cfg
|
| 9 |
from detectron2 import model_zoo
|
|
|
|
|
|
|
| 10 |
|
| 11 |
# Output directory
|
| 12 |
output_dir = "key/"
|
|
|
|
| 4 |
import json
|
| 5 |
import os
|
| 6 |
import gradio as gr
|
| 7 |
+
from detectron2.detectron2.engine import DefaultPredictor
|
| 8 |
+
from detectron2.detectron2.config import get_cfg
|
| 9 |
from detectron2 import model_zoo
|
| 10 |
+
import torch_utils
|
| 11 |
+
import dnnlib
|
| 12 |
|
| 13 |
# Output directory
|
| 14 |
output_dir = "key/"
|