Cleaned imports for find_similar_images.py