venkat-natchi's picture
Update image_generator.py
751f9c2