In reload_index(), we basically get all the documents with all their infos. Within this method, the predicted labels can easily be accessed with result['predicted_label'] --> reload_index() can easily pass the predicted_labels as argument of __inst_doc_from_id()
 (hmm, if you do so, it could also be a good idea to rename __inst_doc_from_id() into __inst_doc() ... :)