add features

app.py

@@ -18,7 +18,7 @@ model = imagebind_model.imagebind_huge(pretrained=True)
 model.eval()
 model.to(device)
 
-image_features = pickle.load(open("./assets/image_features_norm.pkl","rb"))
+image_features = pickle.load(open("./assets/image_features_norm_2.pkl","rb"))
 image_paths = pickle.load(open("./assets/image_paths.pkl","rb"))
 
 

assets/image_features_norm_2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:414236d649565bf26c4dfaac062886f7c73ba60d07029a52f2eb44aa31ff2a38
+size 33172

utils.py

@@ -1,11 +0,0 @@
-import os
-import pickle
-
-# teste = pickle.load(open("df_to_search.pkl","rb"))
-def calculate_similarity():
-    image_features = embeddings[ModalityType.VISION]
-    text_features = embeddings[ModalityType.TEXT]
-
-    image_features /= image_features.norm(dim=-1, keepdim=True)
-    text_features /= text_features.norm(dim=-1, keepdim=True)
-    similarity = text_features.cpu().numpy() @ image_features.cpu().numpy().T