label_mapping.py

index_to_label = {
    0: 'battery',
    1: 'biological',
    2: 'cardboard',
    3: 'clothes',
    4: 'glass',
    5: 'metal',
    6: 'paper',
    7: 'plastic',
    8: 'shoes',
    9: 'trash',
}

label_to_index = {v: k for k, v in index_to_label.items()}