Update image_search.py

image_search.py

@@ -45,7 +45,7 @@ def process_text(caption, tokenizer):
 
 def search_images(search_text, image_path, k = 1):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    model, tokenizer = load_model(device, model_path="models/clip_bangla.pt")
+    model, tokenizer = load_model(device, model_path="clip_bangla.pt")
     image_filenames = glob(image_path + "/*.jpg") + glob(image_path + "/*.JPEG") + glob(image_path + "/*.JPG") + glob(image_path + "/*.png") + glob(image_path + "/*.bmp")
     print(f"Searching in image database >> {image_filenames}")
     imgs = process_image(image_filenames)