Робот на python

Имеется программа в языке Python на предмет искусственного интеллекта.Тема: Умный робот на складеНужно помочь имплементировать уже написанную часть распознавания цифр на картинках в конкретные действия робота (отвезти "пачку" к определенной полке, в зависимости от числа)neural_network.py - распознавание чиселЗапускается программа, генерируется число, понимает, от условия едет к полке и сноваМогу подробнее объяснить кодРассмотрю любые предложения