Free memory with the pool, add in predict task