Spaces:
Running
on
Zero
Running
on
Zero
add hand dataset
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@
|
|
3 |
import os
|
4 |
USE_HUGGINGFACE_ZEROGPU = os.getenv("USE_HUGGINGFACE_ZEROGPU", "False").lower() in ["true", "1", "yes"]
|
5 |
DOWNLOAD_ALL_MODELS_DATASETS = os.getenv("DOWNLOAD_ALL_MODELS_DATASETS", "False").lower() in ["true", "1", "yes"]
|
6 |
-
|
7 |
if USE_HUGGINGFACE_ZEROGPU: # huggingface ZeroGPU, dynamic GPU allocation
|
8 |
try:
|
9 |
import spaces
|
@@ -39,6 +39,7 @@ DATASET_TUPS = [
|
|
39 |
('Borismile/Anime-dataset', None),
|
40 |
('Multimodal-Fatima/CUB_train', 200),
|
41 |
('mrm8488/ImageNet1K-val', 1000),
|
|
|
42 |
]
|
43 |
DATASET_NAMES = [tup[0] for tup in DATASET_TUPS]
|
44 |
DATASET_CLASSES = [tup[1] for tup in DATASET_TUPS]
|
|
|
3 |
import os
|
4 |
USE_HUGGINGFACE_ZEROGPU = os.getenv("USE_HUGGINGFACE_ZEROGPU", "False").lower() in ["true", "1", "yes"]
|
5 |
DOWNLOAD_ALL_MODELS_DATASETS = os.getenv("DOWNLOAD_ALL_MODELS_DATASETS", "False").lower() in ["true", "1", "yes"]
|
6 |
+
|
7 |
if USE_HUGGINGFACE_ZEROGPU: # huggingface ZeroGPU, dynamic GPU allocation
|
8 |
try:
|
9 |
import spaces
|
|
|
39 |
('Borismile/Anime-dataset', None),
|
40 |
('Multimodal-Fatima/CUB_train', 200),
|
41 |
('mrm8488/ImageNet1K-val', 1000),
|
42 |
+
("trashsock/hands-images", 2000),
|
43 |
]
|
44 |
DATASET_NAMES = [tup[0] for tup in DATASET_TUPS]
|
45 |
DATASET_CLASSES = [tup[1] for tup in DATASET_TUPS]
|