Prediction pranavsekhar/animal-to-train