We are working on developing a surrogate model for predicting movement of particles in flows. The work is based using a cut finite element methods that can describe accurately the behavior of drops in a flow.