Fix limiting logits to ind_to_filter (removed this in get_boxes_from_prediction)

app.py

@@ -245,13 +245,9 @@ def preprocess(transform, image, input_prompts = None):
     return input_image, input_image_exemplar, exemplar
 
 def get_boxes_from_prediction(model_output, text, keywords = ""):
-    ind_to_filter = get_ind_to_filter(text, model_output["token"][0].word_ids, keywords)
-    logits = model_output["pred_logits"].sigmoid()[0][:, ind_to_filter]
+    logits = model_output["pred_logits"].sigmoid()[0][:, :]
     boxes = model_output["pred_boxes"][0]
-    if len(keywords.strip()) > 0:
-        box_mask = (logits > CONF_THRESH).sum(dim=-1) == len(ind_to_filter)
-    else:
-        box_mask = logits.max(dim=-1).values > CONF_THRESH
+    box_mask = logits.max(dim=-1).values > CONF_THRESH
     boxes = boxes[box_mask, :].cpu().numpy()
     logits = logits[box_mask, :].cpu().numpy()
     return boxes, logits