default tag count to 0 if not found

hydra_node/models.py

@@ -745,7 +745,7 @@ class EmbedderModel(nn.Module):
         D, I = D.squeeze(), I.squeeze()
         for i, id in enumerate(I):
             tag = index[id]
-            count = tag_count[tag]
+            count = 0 if tag not in tag_count else tag_count[tag]
             prob = D[i]
             results.append([tag, count, prob])