cyberagent
/

markupdm

Text Generation

design completion

markup document

Model card Files Files and versions

ktrk115 commited on Jan 7

Commit

0012b8d

·

verified ·

1 Parent(s): 6d6c47a

Update processing_markupdm.py

Files changed (1) hide show

processing_markupdm.py +1 -1

processing_markupdm.py CHANGED Viewed

@@ -431,7 +431,7 @@ class MarkupDMProcessor(ProcessorMixin):  # type: ignore
         assert matched is not None, "SVG tag not found"
         i = matched.span()[1]
         style = f"<style>{style_text}</style>"
-        example["svg"] = svg[:i] + style.as_str() + svg[i:]
     def render(self, example: dict, save_dir: str | Path | None = None) -> Image.Image:
         with tempfile.TemporaryDirectory() as tmp_dir:

         assert matched is not None, "SVG tag not found"
         i = matched.span()[1]
         style = f"<style>{style_text}</style>"
+        example["svg"] = svg[:i] + style + svg[i:]
     def render(self, example: dict, save_dir: str | Path | None = None) -> Image.Image:
         with tempfile.TemporaryDirectory() as tmp_dir: