Update get_similiarty.py

get_similiarty.py

@@ -4,7 +4,7 @@ import clip
 import torch
 def get_similiarity(prompt, model_resnet, model_vit,  top_k=3):
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    data_dir = 'sample/data'
+    data_dir = 'sample/sample/data'
     image_arr = np.loadtxt("embeddings.csv", delimiter=",")
     raw_dataset = datasets.ImageFolder(data_dir)
     # получите список всех изображений