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:
|