Spaces:
Runtime error
Runtime error
Commit
·
d502bf1
1
Parent(s):
5daa835
Update requirements.txt
Browse files- requirements.txt +7 -3
requirements.txt
CHANGED
|
@@ -1,10 +1,14 @@
|
|
| 1 |
-
|
| 2 |
-
torch
|
|
|
|
| 3 |
torchvision==0.8.2
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
streamlit==1.8.1
|
| 5 |
st-clickable-images==0.0.3
|
| 6 |
func-timeout==4.3.5
|
| 7 |
scikit-learn==1.0.2
|
| 8 |
scikit-image==0.19.2
|
| 9 |
imageio==2.17.0
|
| 10 |
-
torch-geometric==1.7.2
|
|
|
|
| 1 |
+
-f https://download.pytorch.org/whl/cpu/torch_stable.html
|
| 2 |
+
-f https://data.pyg.org/whl/torch-1.10.0+cu113.html
|
| 3 |
+
torch==1.7.1+cu102
|
| 4 |
torchvision==0.8.2
|
| 5 |
+
torch-scatter==2.0.8
|
| 6 |
+
torch-sparse==0.6.11
|
| 7 |
+
torch-geometric
|
| 8 |
+
opencv-python-headless
|
| 9 |
streamlit==1.8.1
|
| 10 |
st-clickable-images==0.0.3
|
| 11 |
func-timeout==4.3.5
|
| 12 |
scikit-learn==1.0.2
|
| 13 |
scikit-image==0.19.2
|
| 14 |
imageio==2.17.0
|
|
|