Fix names

app.py

@@ -57,8 +57,8 @@ def get_scores(crops: List[PIL.Image.Image], query: str) -> torch.Tensor:
     txt_features = model.encode_text(token)
     img_features /= img_features.norm(dim=-1, keepdim=True)
     txt_features /= txt_features.norm(dim=-1, keepdim=True)
-    probs = 100.0 * img_features @ txt_features.T
-    return probs[:, 0].softmax(dim=0)
+    similarity = (100.0 * img_features @ txt_features.T).softmax(dim=0)
+    return similarity
 
 
 def filter_masks(
@@ -82,7 +82,7 @@ def filter_masks(
         filtered_masks.append(mask)
 
         x, y, w, h = mask["bbox"]
-        crop = image[y : y + h, x : x + w]
+        crop = image[y: y + h, x: x + w]
         crop = cv2.cvtColor(crop, cv2.COLOR_BGR2RGB)
         crop = PIL.Image.fromarray(np.uint8(crop * 255)).convert("RGB")
         crop.resize((CLIP_WIDTH, CLIP_HEIGHT))