When loading cards, recognize image sections

utils.py

@@ -9,7 +9,12 @@ from dataclasses import dataclass
 from pathlib import Path
 
 from skops import card
-from skops.card._model_card import Section
+from skops.card._model_card import PlotSection, Section
+
+
+PAT_MD_IMG = re.compile(
+    r'(!\[(?P<image_title>[^\]]+)\]\((?P<image_path>[^\)"\s]+)\s*([^\)]*)\))'
+)
 
 
 def get_rendered_model_card(model_card: card.Card, hf_path: str) -> str:
@@ -42,10 +47,7 @@ def process_card_for_rendering(rendered: str) -> tuple[str, str]:
     def markdown_images(markdown):
         # example image markdown:
         # ![Test image](images/test.png "Alternate text")
-        images = re.findall(
-            r'(!\[(?P<image_title>[^\]]+)\]\((?P<image_path>[^\)"\s]+)\s*([^\)]*)\))',
-            markdown,
-        )
+        images = PAT_MD_IMG.findall(markdown)
         return images
 
     def img_to_bytes(img_path):
@@ -106,11 +108,20 @@ def iterate_key_section_content(
             continue
 
         return_key = key if not parent_keys else "/".join(parent_keys + [key])
+        content = val.content
+
         is_fig = getattr(val, "is_fig", False)
+        img_match = PAT_MD_IMG.match(val.content)
+        if img_match:  # image section found in parsed model card
+            is_fig = True
+            img_title = img_match.groupdict()["image_title"]
+            img_path = img_match.groupdict()["image_path"]
+            content = PlotSection(alt_text=img_title, path=img_path)
+
         yield SectionInfo(
             return_key=return_key,
             title=title,
-            content=val.content,
+            content=content,
             is_fig=is_fig,
             level=level,
         )