Update src/app.py

src/app.py

@@ -65,7 +65,7 @@ def get_neighbors(query_image, selected_descriptor, top_k=5):
     Returns:
         A list of the top_k most similar images as PIL objects.
     """
-    if  "Color Descriptor" in selected_descriptor:
+    if  "Color Descriptor" == selected_descriptor:
         cd = ColorDescriptor((8, 12, 3))
         qi_embedding = cd.describe(query_image)
         qi_np = np.array(qi_embedding)
@@ -73,14 +73,14 @@ def get_neighbors(query_image, selected_descriptor, top_k=5):
             'color_embeddings', qi_np, k=top_k)
         images = retrieved_examples['image'] #retrieved images is a dict, with images and embeddings
         return images
-    if "CLIP" in selected_descriptor:
+    if "CLIP" == selected_descriptor:
         clip_model = CLIPImageEncoder()
         qi_embedding = clip_model.encode_image(query_image)
         scores, retrieved_examples = dataset_with_embeddings.get_nearest_examples(
             'clip_embeddings', qi_embedding, k=top_k)
         images = retrieved_examples['image']
         return images
-    if "LBP" in selected_descriptor:
+    if "LBP" == selected_descriptor:
         lbp_model = LBPImageEncoder(8,2)
         qi_embedding = lbp_model.describe(query_image)
         scores, retrieved_examples = dataset_with_embeddings.get_nearest_examples(
@@ -117,14 +117,14 @@ with gr.Blocks() as demo:
     """)
 
     with gr.Row():
+        descr_dropdown = gr.Dropdown(["Color Descriptor", "LBP", "CLIP"], value="LBP", label="Please choose an descriptor")
         checkboxes_descr = gr.CheckboxGroup(["Color Descriptor", "LBP", "CLIP"], label="Please choose an descriptor")
         dataset_dropdown = gr.Dropdown(
         ["huggan/CelebA-faces", "EIT/cbir-eit"],
-        value=["huggan/CelebA-faces"]
+        value="huggan/CelebA-faces",
+        label="Please select a dataset"
     )
-    btn_index = gr.Button(value="Switch Dataset")
-    btn_index.click(load_cbir_dataset, inputs=[dataset_dropdown])
-    btn.click(get_neighbors, inputs=[image_input, checkboxes_descr], outputs=[gallery_output])
+    btn.click(get_neighbors, inputs=[image_input, descr_dropdown], outputs=[gallery_output])
 
     # gr.Markdown(
     # """