app/config.py

import pandas as pd

class CFG:
    LABEL_PATH = 'data/labels.csv'

    labels = pd.read_csv(LABEL_PATH)
    idx_to_class = dict(enumerate(labels.breed.unique()))
    class_to_idx = {c:i for i,c in idx_to_class.items()}


    # Model related
    MODEL = 'inception_v4'
    PRETRAINED = True
    IMG_SIZE = (299, 299) # Depends in base model