Spaces:
Sleeping
Sleeping
Added weight
Browse files- .gitignore +0 -1
- model1.pt +3 -0
- 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('
|
| 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 |
|