huzey commited on
Commit
5c0c30a
1 Parent(s): 96f34e8

add hand dataset

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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]