Bom dia Gabriel.
Tentei resolver o erro revendo todo o código, mas o erro persiste.
Estou colocando o link aqui pra que veja se pode me ajudar.
https://colab.research.google.com/drive/1fPNm32aGNs0lPahmgxULDOQOnLP-xBmH?usp=sharing
Agradeço antecipadamente.