Bert Model Implementation Python