checkpoints test.py model.py visual_prompting.py test.png output.png __pycache__