Map formula prompt key to the equation block type

app.py

@@ -104,6 +104,8 @@ REGION_TASKS = [
     ("Chart / scientific figure \u2192 table", "char"),
     ("Seal", "seal"),
 ]
+# Prompt keys and block-type names differ for formulas ("formula" vs "equation").
+TASK_BLOCK_TYPES = {"formula": "equation"}
 # The model prefixes recognized code with its own language marker, e.g. `<_Python_>`.
 CODE_LANG_RE = re.compile(r"^\s*<_([A-Za-z0-9+#._\- ]+)_>\s*")
 
@@ -297,7 +299,11 @@ def parse_document(
             DEFAULT_PROMPTS[task],
             _sampling_params(task, max_new_tokens),
         )
-        block = ContentBlock(type=task, bbox=[[0.0, 0.0], [1.0, 1.0]], content=output)
+        block = ContentBlock(
+            type=TASK_BLOCK_TYPES.get(task, task),
+            bbox=[[0.0, 0.0], [1.0, 1.0]],
+            content=output,
+        )
         blocks = helper.post_process([block]) or [block]
         raw_markdown, display_markdown = blocks_to_markdown(page, blocks, False)
         seconds = time.time() - started