fix: fixed prompt of code formula predictor (#72)

Signed-off-by: Matteo Omenetti <[email protected]>
DS4SD · Jan 22, 2025 · bdcc82f · bdcc82f
1 parent b01f90d
commit bdcc82f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docling_ibm_models/code_formula_model/code_formula_predictor.py b/docling_ibm_models/code_formula_model/code_formula_predictor.py
@@ -119,7 +119,7 @@ def _get_prompt(self, label: str) -> str:
         prompt = (
             "A chat between a curious user and an artificial intelligence"
             " assistant. The assistant gives helpful, detailed, and polite answers to"
-            " the user's questions. USER:"
+            " the user's questions. USER: "
         )
         prompt += (
             "<img>" + "<imgpad>" * 256 + "</img>" + "\n" + " ASSISTANT:" + "\n" + query