Update utils/tools.py

utils/tools.py

@@ -419,7 +419,11 @@ def text_prompt(annotations, text, img_path, device, wider=False, threshold=0.9)
     cropped_boxes, cropped_images, not_crop, origin_id, annotations_ = crop_image(
         annotations, img_path
     )
-    clip_model, preprocess = clip.load("./weights/CLIP_ViT_B_32.pt", device=device)
+    import os 
+    clip_model_path = "CLIP_ViT_B_32.pt"
+    assert os.path.exists(clip_model_path)
+    clip_model, preprocess = clip.load(clip_model_path, device=device)
+    #clip_model, preprocess = clip.load("./weights/CLIP_ViT_B_32.pt", device=device)
     scores = retriev(
         clip_model, preprocess, cropped_boxes, text, device=device
     )