diff --git a/docling_ibm_models/code_formula_model/code_formula_predictor.py b/docling_ibm_models/code_formula_model/code_formula_predictor.py index 52849a7..5804558 100644 --- 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