Add models

model1.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:13d24332c79fb809ab40a9af19e64e018b428a918f5539324a4815c4685a5ff4
+size 1264309243

similarity_search.py

@@ -89,9 +89,9 @@ def get_feature_vector_img(model, imgs, epoch=10, use_cuda=False):
     return np.concatenate(features, axis=0)
 
 def Model():
-    # backbone = open_clip.create_model_and_transforms('ViT-H-14', None)[0].visual
-    # backbone.load_state_dict(th.load('./model1.pt'))  #https://huggingface.co/hca97/aicrowd-visual-recognition-models/blob/main/model1.pt
-    backbone, preprocess_train, preprocess_val = open_clip.create_model_and_transforms('hf-hub:Thenujan/ViT-H-14')
+    backbone = open_clip.create_model_and_transforms('ViT-H-14', None)[0].visual
+    backbone.load_state_dict(th.load('./model1.pt'))  #https://huggingface.co/hca97/aicrowd-visual-recognition-models/blob/main/model1.pt
+    # backbone, preprocess_train, preprocess_val = open_clip.create_model_and_transforms('hf-hub:Thenujan/ViT-H-14')
     backbone.eval()   # Dropping unecessary layers
     return backbone