Update predict_tags.py

predict_tags.py

@@ -246,16 +246,13 @@ def predict_tag(node, sibling_count, prev_sibling_tag, parent_height, parent_bg_
     if figma_type == "TEXT":
         node["tag"] = "P"
         node["base_tag"] = "P"
-    elif figma_type == "SVG" or node.get("tag") == "SVG" or ((fills := node_data.get("fills", [])) and any(fill.get("type") == "IMAGE" for fill in fills)):
+    elif figma_type == "VECTOR" or figma_type == "SVG" or node.get("tag") == "SVG" or ((fills := node_data.get("fills", [])) and any(fill.get("type") == "IMAGE" for fill in fills)):
         node["tag"] = "SVG"
         node["base_tag"] = "SVG"
         w = node.get("node", {}).get("width", 0)
         h = node.get("node", {}).get("height", 0)
         if w * h < 2000:
             node["tag"] = "ICON"
-    elif figma_type == "VECTOR":
-        node["tag"] = "ICON"
-        node["base_tag"] = "ICON"
     elif figma_type == "LINE":
         node["tag"] = "HR"
         node["base_tag"] = "HR"