todo for me
gotta update this card later
update the dataset with validation pics (and possibly test) and train this model again
figure out why i cant run it on vscode
