add common name

app.py

@@ -115,6 +115,13 @@ def zero_shot_classification(img, cls_str: str) -> dict[str, float]:
     return {cls: prob for cls, prob in zip(classes, probs)}
 
 
+def format_name(taxon, common):
+    taxon = " ".join(taxon)
+    if not common:
+        return taxon
+    return f"{taxon} ({common})"
+
+
 @torch.no_grad()
 def open_domain_classification(img, rank: int) -> dict[str, float]:
     """
@@ -133,13 +140,13 @@ def open_domain_classification(img, rank: int) -> dict[str, float]:
     if rank + 1 == len(ranks):
         topk = probs.topk(k)
         return {
-            " ".join(txt_names[i]): prob for i, prob in zip(topk.indices, topk.values)
+            format_name(*txt_names[i]): prob for i, prob in zip(topk.indices, topk.values)
         }
 
     # Sum up by the rank
     output = collections.defaultdict(float)
     for i in torch.nonzero(probs > min_prob).squeeze():
-        output[" ".join(txt_names[i][: rank + 1])] += probs[i]
+        output[" ".join(txt_names[i][0][: rank + 1])] += probs[i]
 
     topk_names = heapq.nlargest(k, output, key=output.get)
 

txt_emb_species.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c71babd1b7bc275a1dbb12fd36e6329bcc2487784c0b7be10c2f4d0031d34211
-size 50445969
+oid sha256:844e6fabc06cac072214d566b78f40825b154efa9479eb11285030ca038b2ece
+size 65731052