I built something similar for images. It's command line based, so one could simply create a bash script to copy data to a folder based on a prediction.

https://github.com/vergeml/vergeml