huzey commited on
Commit
4d58f0c
1 Parent(s): 0daddbf

fix dataset

Browse files
Files changed (1) hide show
  1. app.py +2 -4
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 = list(map(int, filter_by_class_text.split(",")))
 
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]