[FEATURE] - Implement proper generation logging message

Right now we have a somewhat non-informative message "generated and saved X samples".
→ Consider a proper SUCCESS, WARNING, ERROR logging with a few details, at leas the model (and the generation time it took). 
→ Use loguru