abstract mapping of class idx to class names
jeremyjordan opened this issue · comments
What's your idea?
Currently, our Streamlit app uses a hardcode list for mapping the model prediction to human-readable names. As we support new models trained on datasets with potentially different sets of class names (e.g. we might decide to add flower breeds), we'll want to be more flexible in how we perform this mapping.
Describe the solution you'd like
A clear and concise description of what you want to happen.
Describe alternatives you've considered
A clear and concise description of any alternative solutions or features you've considered.