Update current.py

Optimizing condition of image format