It can be fixed with inpaint but the machine it was running out of time. So, I could work on more details, that's one of the biggest problems on free plans like colab or paperspace.