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
|