Update image_cap.py