huzey commited on
Commit
2fbda21
1 Parent(s): d90f66b

fix egothink

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -56,11 +56,11 @@ DATASETS = {
56
  ('huggan/anime-faces', None),
57
  ],
58
  'Pose': [
59
- ('razdab/sign_pose_M', None),
60
  ('sayakpaul/poses-controlnet-dataset', None),
61
  ('EgoThink/EgoThink', None),
62
- ('junjuice0/vtuber-tachi-e', None),
63
  ('Fiacre/small-animal-poses-controlnet-dataset', None),
 
64
  ],
65
  'Hand': [
66
  ('trashsock/hands-images', 8),
@@ -1450,7 +1450,10 @@ def make_input_images_section(rows=1, cols=3, height="auto", advanced=False, is_
1450
  return default_images
1451
  try:
1452
  progress(0.5, desc="Downloading Dataset")
1453
- dataset = load_dataset(dataset_name, trust_remote_code=True)
 
 
 
1454
  key = list(dataset.keys())[0]
1455
  dataset = dataset[key]
1456
  except Exception as e:
 
56
  ('huggan/anime-faces', None),
57
  ],
58
  'Pose': [
 
59
  ('sayakpaul/poses-controlnet-dataset', None),
60
  ('EgoThink/EgoThink', None),
61
+ ('razdab/sign_pose_M', None),
62
  ('Fiacre/small-animal-poses-controlnet-dataset', None),
63
+ ('junjuice0/vtuber-tachi-e', None),
64
  ],
65
  'Hand': [
66
  ('trashsock/hands-images', 8),
 
1450
  return default_images
1451
  try:
1452
  progress(0.5, desc="Downloading Dataset")
1453
+ if 'EgoThink' in dataset_name:
1454
+ dataset = load_dataset(dataset_name, 'Activity', trust_remote_code=True)
1455
+ else:
1456
+ dataset = load_dataset(dataset_name, trust_remote_code=True)
1457
  key = list(dataset.keys())[0]
1458
  dataset = dataset[key]
1459
  except Exception as e: