Update image_search.py

image_search.py

@@ -418,3 +418,10 @@ def load_custom_model(model_dir=".", base_image_dir="flickr8k/Flicker8k_Dataset"
             fixed_paths.append(path)
 
     return nn_model, image_embeddings, fixed_paths
+
+# Inside image_search.py
+
+def update_custom_image_paths(folder_path):
+    global custom_image_paths
+    custom_image_paths = [os.path.join(folder_path, name) for name in os.listdir(folder_path) if name.lower().endswith((".jpg", ".jpeg", ".png"))]
+