liangyuch commited on
Commit
3c2f5f5
β€’
1 Parent(s): faed248

detectron2 cu102

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,7 +15,8 @@ if os.getenv('SYSTEM') == 'spaces':
15
  subprocess.call('pip uninstall -y opencv-python-headless'.split())
16
  subprocess.call('pip install opencv-python-headless==4.5.5.64'.split())
17
  subprocess.call('pip install pycocotools'.split())
18
- subprocess.call("pip install detectron2==0.6 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html".split())
 
19
  # detectron2==0.6 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html
20
  import cv2
21
  import gradio as gr
 
15
  subprocess.call('pip uninstall -y opencv-python-headless'.split())
16
  subprocess.call('pip install opencv-python-headless==4.5.5.64'.split())
17
  subprocess.call('pip install pycocotools'.split())
18
+ subprocess.call("pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html".split())
19
+ # pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html
20
  # detectron2==0.6 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html
21
  import cv2
22
  import gradio as gr