diff --git a/src/controllers/suggest_controller.py b/src/controllers/suggest_controller.py index 204017d7a8215aa2bb7519a2cf6d1fcec8f7e88f..25aa1b250047bc916961ba9e58f0d2e0acc304a6 100644 --- a/src/controllers/suggest_controller.py +++ b/src/controllers/suggest_controller.py @@ -11,7 +11,7 @@ class SuggestController: client = Client("mmcgovern574/Mistral-Small-24B-Instruct-2501", hf_token) result = client.predict( - message=message, + message=f'Enhance this prompt. The prompt will be used by an Image2Image model (only give one answer, don\'t add anything else) : {message}', param_2=system_prompt, param_3=temperature, param_4=max_new_tokens,