Spaces:
Runtime error
Runtime error
Update requirements.txt
Browse files- requirements.txt +5 -2
requirements.txt
CHANGED
|
@@ -2,8 +2,11 @@ opencv-python-headless
|
|
| 2 |
pandas
|
| 3 |
numpy
|
| 4 |
matplotlib
|
| 5 |
-
|
|
|
|
|
|
|
|
|
|
| 6 |
# Detectron
|
| 7 |
-
--find-links https://dl.fbaipublicfiles.com/detectron2/wheels/
|
| 8 |
detectron2
|
| 9 |
grad-cam
|
|
|
|
| 2 |
pandas
|
| 3 |
numpy
|
| 4 |
matplotlib
|
| 5 |
+
# Torch
|
| 6 |
+
--find-links https://download.pytorch.org/whl/cpu/torch_stable.html
|
| 7 |
+
torch==1.10.1+cpu
|
| 8 |
+
torchvision==0.11.2+cpu
|
| 9 |
# Detectron
|
| 10 |
+
--find-links https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.10/index.html
|
| 11 |
detectron2
|
| 12 |
grad-cam
|