dataset fix

app.py

@@ -5,7 +5,7 @@ import clip
 import torch
 
 
-data = pickle.load(open("embeddings.pkl", "rb"))
+data = pickle.load(open("embeddings_nothumb.pkl", "rb"))
 embeddings = data["embeddings"]
 image_features = torch.Tensor(embeddings)
 image_features /= image_features.norm(dim=-1, keepdim=True)
@@ -14,7 +14,6 @@ image_features /= image_features.norm(dim=-1, keepdim=True)
 n, d = embeddings.shape
 
 filenames = data["filenames"]
-thumbs = data["thumbs"]
 
 base_url = "https://static.renyi.hu/ai-shared/daniel/sameenergy/index/"
 urls = [base_url + filename for filename in filenames]