karolmajek commited on
Commit
0a02fce
·
1 Parent(s): 82da21b

lazy install detectron2

Browse files
Files changed (2) hide show
  1. app.py +6 -0
  2. requirements.txt +0 -1
app.py CHANGED
@@ -1,3 +1,9 @@
 
 
 
 
 
 
1
  from matplotlib.pyplot import axis
2
  import gradio as gr
3
  import requests
 
1
+ try:
2
+ import detectron2
3
+ except:
4
+ import os
5
+ os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
6
+
7
  from matplotlib.pyplot import axis
8
  import gradio as gr
9
  import requests
requirements.txt CHANGED
@@ -1,4 +1,3 @@
1
  opencv-python-headless
2
  pyyaml==5.1
3
- git+https://github.com/facebookresearch/detectron2.git
4
  torch
 
1
  opencv-python-headless
2
  pyyaml==5.1
 
3
  torch