what notebook did you train your model on