Update src/app.py

src/app.py

@@ -1,7 +1,7 @@
 from colordescriptor import ColorDescriptor
 from CLIP import CLIPImageEncoder
 from LBP import LBPImageEncoder
-from helper import chi2_distance, euclidean_distance
+from helper import chi2_distance, euclidean_distance, merge_features
 import gradio as gr
 import os
 import cv2
@@ -28,6 +28,9 @@ def index_dataset(dataset):
     clip_model = CLIPImageEncoder()
     dataset_with_embeddings = dataset_with_embeddings.map(clip_model.encode_images, batched=True, batch_size=16)
 
+    print("LBP and Color")
+    dataset_with_embeddings = dataset_with_embeddings.map(lambda row: {'lbp_color_embeddings': merge_features(row['lbp_embeddings'], row['color_embeddings']})
+
 
     # Add index
     dataset_with_embeddings.add_faiss_index(column='color_embeddings')