Yeah that's literally it, still misses a bit as my dataset wasn't that large nor good (idk where to get real-life images for this lol), but posting it anyway. Use milking machine somewhere in the prompt.
milking machine
Initial version
milking_machine.safetensors