From d22f6b12d23788c1dea8a68a65383aa31fc9ddb3 Mon Sep 17 00:00:00 2001
From: BIRK Renaud <renaud.birk@imt-atlantique.net>
Date: Tue, 25 Feb 2025 15:41:39 +0100
Subject: [PATCH] fix: suggest controller to enhance prompt
---
src/controllers/suggest_controller.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/controllers/suggest_controller.py b/src/controllers/suggest_controller.py
index 204017d..25aa1b2 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,
--
GitLab