i change from (in main.py): if config.labels_to_names[label] not in curr_label_list: to: if config.labels_to_names[label] in curr_label_list: