import jsonfrom transformers import CLIPProcessor, CLIPTextModelWithProjectionfrom torch import load, matmul, argsortfrom torch.nn.functional import softmax