MattyWhite commited on
Commit
8b45a4d
·
verified ·
1 Parent(s): c3d8b81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -1,5 +1,9 @@
1
  import os
2
- os.system('pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html')
 
 
 
 
3
  import gradio as gr
4
  import torch
5
  import cv2
 
1
  import os
2
+ try:
3
+ import detectron2
4
+ except:
5
+ import os
6
+ os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
7
  import gradio as gr
8
  import torch
9
  import cv2