Chnaged input type

similarity_search.py

@@ -90,18 +90,20 @@ def get_feature_vector_img(model, imgs, epoch=10, use_cuda=False):
 
 def Model():
     backbone = open_clip.create_model_and_transforms('ViT-H-14', None)[0].visual
-    backbone.load_state_dict(th.load('https://huggingface.co/Thenujan/ViT-H-14/blob/main/model.bin'))  # https://huggingface.co/hca97/aicrowd-visual-recognition-models/blob/main/model1.pt
+    backbone.load_state_dict(th.load('https://huggingface.co/Thenujan/ViT-H-14/blob/main/model.bin'))  #https://huggingface.co/hca97/aicrowd-visual-recognition-models/blob/main/model1.pt
     backbone.eval()   # Dropping unecessary layers
     return backbone
 
 def find(image):
+    # img_path = '/kaggle/input/vprtestdata/public_dataset/queries/abiding-inchworm-of-ultimate-freedom.jpeg'
+
+    # image = read_image(img_path)
+    image = np.array(image)
 
     image = transform_img(image)
 
     image = image.unsqueeze(dim=0)
 
-    # print(image.shape)
-
     feature_vectors_gallery = np.load('./VPR image similarity search/gallary_embedding/feature_vectors_gallery.npy')
 
     model = Model()