Update src/app.py

src/app.py

@@ -13,18 +13,20 @@ dataset.cleanup_cache_files()
 candidate_subset = dataset["train"].select(range(10)) # This is a small CBIR app! :D
 
 def index_dataset(dataset):
+    print(dataset)
+
+    print("LBP Embeddings")
+    lbp_model = LBPImageEncoder(8,2)
+    dataset_with_embeddings = dataset.map(lambda row: {'lbp_embeddings': lbp_model.describe(row["image"])})
 
     print("Color Embeddings")
     cd = ColorDescriptor((8, 12, 3))
-    dataset_with_embeddings = dataset.map(lambda row: {'color_embeddings': cd.describe(row["image"])})
+    dataset_with_embeddings = dataset_with_embeddings.map(lambda row: {'color_embeddings': cd.describe(row["image"])})
 
     print("CLIP Embeddings")
     clip_model = CLIPImageEncoder()
     dataset_with_embeddings = dataset_with_embeddings.map(clip_model.encode_images, batched=True, batch_size=16)
 
-    print("LBP Embeddings")
-    lbp_model = LBPImageEncoder(8,2)
-    dataset_with_embeddings = dataset_with_embeddings.map(lambda row: {'lbp_embeddings': lbp_model.describe(row["image"])})
 
     # Add index
     dataset_with_embeddings.add_faiss_index(column='color_embeddings')