Merged dataset

src/app.py

@@ -98,14 +98,20 @@ def get_neighbors(query_image, selected_descriptor, selected_distance, top_k=5):
             'lbp_embeddings', qi_embedding, k=top_k)
         images = retrieved_examples['image']
         return images
+    if "LBPColor" == selected_descriptor:
+        lbp_model = LBPImageEncoder(8,2)
+        cd = ColorDescriptor((8, 12, 3))
+        qi_embedding = merge_features(lbp_model.describe(query_image), cd.describe(query_image))
+        scores, retrieved_examples = dataset_with_embeddings.get_nearest_examples(
+            'lbp_color_embeddings', qi_embedding, k=top_k
+        )
+
+
     else:
         print("This descriptor is not yet supported :(")
         return []
 
 
-def load_cbir_dataset(datasetname, size=1000):
-    pass
-
 # Define the Gradio Interface
 
 with gr.Blocks() as demo:
@@ -130,11 +136,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         descr_dropdown = gr.Dropdown(["Color Descriptor", "LBP", "CLIP"], value="LBP", label="Please choose an descriptor")
         dist_dropdown = gr.Dropdown(["FAISS", "Chi-squared", "Euclid"], value="FAISS", label="Please choose a distance measure")
-        dataset_dropdown = gr.Dropdown(
-        ["huggan/CelebA-faces", "EIT/cbir-eit"],
-        value="huggan/CelebA-faces",
-        label="Please select a dataset"
-    )
+
     btn.click(get_neighbors, inputs=[image_input, descr_dropdown, dist_dropdown], outputs=[gallery_output])