Venue Theme - Display Images for Selected Variation

I have been able to successfully set up the Venue theme with a product with 2 variations (white and grey). However, all of the product images are displayed (both white and grey) regardless of which variation is chosen. I would like to only display the images relevant to the white product when that variation is selected and then switch to the grey product images when the user selects that variation.

There is a tutorial on YouTube with a different codebase, so I’m not sure how to do this.