shivi commited on
Commit
db60804
·
1 Parent(s): afeeaf1

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -2
requirements.txt CHANGED
@@ -1,5 +1,10 @@
1
- torch
 
 
 
 
 
2
  gradio
 
3
  transformers
4
  pillow
5
- detectron2==0.6
 
1
+ --extra-index-url https://download.pytorch.org/whl
2
+ torch==1.9.0+cu111
3
+ torchvision==0.10.0+cu111
4
+ torchaudio==0.9.0
5
+ -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu111/torch1.9/index.html
6
+ detectron2==0.6
7
  gradio
8
+ opencv-python
9
  transformers
10
  pillow