Spaces:
Running
on
Zero
Running
on
Zero
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]
|