please ignore mounting the drive to colaboratory if you are not using google colaboratory Feel free to fork and download the code