Bert Model Implementation Python Code For Snake