basementparking commited on
Commit
fa3a588
·
verified ·
1 Parent(s): 3c5d85e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import gradio as gr
2
  import torch
3
  import numpy as np
@@ -6,6 +9,9 @@ from detectron2.engine import DefaultPredictor
6
  from detectron2.config import get_cfg
7
  from detectron2 import model_zoo
8
  from detectron2.utils.visualizer import Visualizer, ColorMode
 
 
 
9
 
10
  # set Detectron2
11
  cfg = get_cfg()
 
1
+ import os
2
+ os.system("pip install 'git+https://github.com/facebookresearch/detectron2.git'")
3
+
4
  import gradio as gr
5
  import torch
6
  import numpy as np
 
9
  from detectron2.config import get_cfg
10
  from detectron2 import model_zoo
11
  from detectron2.utils.visualizer import Visualizer, ColorMode
12
+ from detectron2.data import MetadataCatalog
13
+
14
+
15
 
16
  # set Detectron2
17
  cfg = get_cfg()