Add predict implementation (read source apply beam etc..)