pierreguillou commited on
Commit
29f6ff3
1 Parent(s): 37c7c40

Update files/functions.py

Browse files
Files changed (1) hide show
  1. files/functions.py +4 -0
files/functions.py CHANGED
@@ -9,6 +9,10 @@ os.system('pip install -q torch==1.10.0+cu111 torchvision==0.11+cu111 -f https:/
9
  #os.system('pip install -q detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.8/index.html')
10
  os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
11
 
 
 
 
 
12
  import gradio as gr
13
  import re
14
  import string
 
9
  #os.system('pip install -q detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.8/index.html')
10
  os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
11
 
12
+ import detectron2
13
+ from detectron2.utils.logger import setup_logger
14
+ setup_logger()
15
+
16
  import gradio as gr
17
  import re
18
  import string