Skip to content
Snippets Groups Projects
Commit d22f6b12 authored by BIRK Renaud's avatar BIRK Renaud
Browse files

fix: suggest controller to enhance prompt

parent 59cec853
No related branches found
No related tags found
No related merge requests found
......@@ -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,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment