Update image_rec.py