Spaces:
Running
on
Zero
Running
on
Zero
fix egothink
Browse files
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 |
-
('
|
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 |
-
|
|
|
|
|
|
|
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:
|