Thenujan commited on
Commit
884b2ca
·
1 Parent(s): c50d8cf

Added weight

Browse files
Files changed (3) hide show
  1. .gitignore +0 -1
  2. model1.pt +3 -0
  3. similarity_search.py +1 -1
.gitignore CHANGED
@@ -1 +0,0 @@
1
- model1.pt
 
 
model1.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13d24332c79fb809ab40a9af19e64e018b428a918f5539324a4815c4685a5ff4
3
+ size 1264309243
similarity_search.py CHANGED
@@ -90,7 +90,7 @@ def get_feature_vector_img(model, imgs, epoch=10, use_cuda=False):
90
 
91
  def Model():
92
  backbone = open_clip.create_model_and_transforms('ViT-H-14', None)[0].visual
93
- 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
94
  backbone.eval() # Dropping unecessary layers
95
  return backbone
96
 
 
90
 
91
  def Model():
92
  backbone = open_clip.create_model_and_transforms('ViT-H-14', None)[0].visual
93
+ backbone.load_state_dict(th.load('./model1.pt')) #https://huggingface.co/hca97/aicrowd-visual-recognition-models/blob/main/model1.pt
94
  backbone.eval() # Dropping unecessary layers
95
  return backbone
96