this function is to show image of adversarial and its probability, I only want to download the image.
def visualize(x, x_adv, x_grad, epsilon, clean_pred, adv