huzey commited on
Commit
7645f3c
1 Parent(s): 5c0c30a

add hand dataset

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -39,7 +39,7 @@ DATASET_TUPS = [
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]
@@ -832,9 +832,10 @@ with demo:
832
  )
833
 
834
  with gr.Tab('Text'):
835
- if USE_HUGGINGFACE_ZEROGPU:
836
  from app_text import make_demo
837
- else:
 
838
  from draft_gradio_app_text import make_demo
839
  make_demo()
840
 
@@ -918,3 +919,7 @@ if DOWNLOAD_ALL_MODELS_DATASETS:
918
  demo.launch(share=True)
919
 
920
  # %%
 
 
 
 
 
39
  ('Borismile/Anime-dataset', None),
40
  ('Multimodal-Fatima/CUB_train', 200),
41
  ('mrm8488/ImageNet1K-val', 1000),
42
+ ("trashsock/hands-images", 8),
43
  ]
44
  DATASET_NAMES = [tup[0] for tup in DATASET_TUPS]
45
  DATASET_CLASSES = [tup[1] for tup in DATASET_TUPS]
 
832
  )
833
 
834
  with gr.Tab('Text'):
835
+ try:
836
  from app_text import make_demo
837
+ except ImportError:
838
+ print("Debugging")
839
  from draft_gradio_app_text import make_demo
840
  make_demo()
841
 
 
919
  demo.launch(share=True)
920
 
921
  # %%
922
+ ds = load_dataset(DATASET_NAMES[-1])
923
+ # %%
924
+ np.unique(ds['train']['label'])
925
+ # %%