For some reason is doesnt work when running in a container despite all the deps being installed. It just crashes and dies in Juypter because this function throws an error on .items():
def id_class_name(class_id, classes):
for key, value in classes.items():
if class_id == key:
return value