image addded

app.py

@@ -49,17 +49,19 @@ def detect_using_clip(image,prompts=[],threshould=0.4):
         # extract countours from the image
         lbl_0 = label(predicted_image)
         props = regionprops(lbl_0)
-        model_detections[prompt.lower()] = [rescale_bbox(prop.bbox,orig_image_shape=image.shape[:2],model_shape=predicted_image.shape[0]) for prop in props]
+        prompt = prompt.lower()
+        model_detections[prompt] = [rescale_bbox(prop.bbox,orig_image_shape=image.shape[:2],model_shape=predicted_image.shape[0]) for prop in props]
 
     return model_detections 
 
 def visualize_images(image,detections,prompt):
     H,W = image.shape[:2]
+    prompt = prompt.lower()
     image_copy = image.copy()
     if prompt not in detections.keys():
         print("prompt not in query ..")
         return image_copy
-    for bbox in detections[prompt.lower()]:
+    for bbox in detections[prompt]:
         cv2.rectangle(image_copy, (int(bbox[1]), int(bbox[0])), (int(bbox[3]), int(bbox[2])), (255, 0, 0), 2)
         cv2.putText(image_copy,str(prompt),(int(bbox[1]), int(bbox[0])),cv2.FONT_HERSHEY_SIMPLEX, 2, 255)
     return image_copy