Spaces:
Sleeping
Sleeping
fix dataset
Browse files
app.py
CHANGED
@@ -536,11 +536,9 @@ def make_dataset_images_section(open=False):
|
|
536 |
if num_images > len(dataset):
|
537 |
num_images = len(dataset)
|
538 |
|
539 |
-
if 'label' not in dataset and is_filter:
|
540 |
-
gr.Error(f"Dataset {dataset_name} has no class label.")
|
541 |
-
return None
|
542 |
if is_filter:
|
543 |
-
classes =
|
|
|
544 |
labels = np.array(dataset['label'])
|
545 |
unique_labels = np.unique(labels)
|
546 |
valid_classes = [i for i in classes if i in unique_labels]
|
|
|
536 |
if num_images > len(dataset):
|
537 |
num_images = len(dataset)
|
538 |
|
|
|
|
|
|
|
539 |
if is_filter:
|
540 |
+
classes = [int(i) for i in filter_by_class_text.split(",")]
|
541 |
+
print(filter_by_class_text, len(classes), classes)
|
542 |
labels = np.array(dataset['label'])
|
543 |
unique_labels = np.unique(labels)
|
544 |
valid_classes = [i for i in classes if i in unique_labels]
|