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

Matteo-Omenetti · web-flow · commit bdcc82f21e5b · 2025-01-22T14:26:46.000+01:00
Signed-off-by: Matteo Omenetti &lt;omenetti.matteo@gmail.com&gt;
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

Original file line number	Diff line number	Diff line change
`@@ -119,7 +119,7 @@ def _get_prompt(self, label: str) -> str:`
`119`	`119`	`prompt = (`
`120`	`120`	`"A chat between a curious user and an artificial intelligence"`
`121`	`121`	`" assistant. The assistant gives helpful, detailed, and polite answers to"`
`122`		`- " the user's questions. USER:"`
	`122`	`+ " the user's questions. USER: "`
`123`	`123`	`)`
`124`	`124`	`prompt += (`
`125`	`125`	`"<img>" + "<imgpad>" * 256 + "</img>" + "\n" + " ASSISTANT:" + "\n" + query`