Model using a dataset of images i generated. I'm still tweaking it and fiddling with the training dataset so yea.