Upload folder using huggingface_hub
Browse files- demo.py +0 -4
- requirements.txt +7 -1
demo.py
CHANGED
@@ -1,7 +1,3 @@
|
|
1 |
-
import os
|
2 |
-
os.system("pip install datasets einops tabulate opencv-python ttach scikit-learn -U &> /dev/null")
|
3 |
-
print("Installation finished.")
|
4 |
-
|
5 |
import os, cv2, time, math
|
6 |
print("=> Loading libraries...")
|
7 |
start = time.time()
|
|
|
|
|
|
|
|
|
|
|
1 |
import os, cv2, time, math
|
2 |
print("=> Loading libraries...")
|
3 |
start = time.time()
|
requirements.txt
CHANGED
@@ -2,4 +2,10 @@ torch>=1.7
|
|
2 |
torchvision
|
3 |
pyyaml
|
4 |
huggingface_hub
|
5 |
-
safetensors>=0.2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
torchvision
|
3 |
pyyaml
|
4 |
huggingface_hub
|
5 |
+
safetensors>=0.2
|
6 |
+
datasets
|
7 |
+
einops
|
8 |
+
tabulate
|
9 |
+
opencv-python
|
10 |
+
ttach
|
11 |
+
scikit-learn
|