Update prompt_expander.py

#2
by YiYiXu HF Staff - opened
Files changed (1) hide show
  1. prompt_expander.py +2 -0
prompt_expander.py CHANGED
@@ -41,6 +41,7 @@ class GeminiPromptExpander(ModularPipelineBlocks):
41
  type_hint=str,
42
  required=True,
43
  description="Prompt to use",
 
44
  )
45
  ]
46
 
@@ -55,6 +56,7 @@ class GeminiPromptExpander(ModularPipelineBlocks):
55
  "prompt",
56
  type_hint=str,
57
  description="Expanded prompt by the LLM",
 
58
  ),
59
  OutputParam(
60
  "old_prompt",
 
41
  type_hint=str,
42
  required=True,
43
  description="Prompt to use",
44
+ metadata={"mellon":"textbox"}
45
  )
46
  ]
47
 
 
56
  "prompt",
57
  type_hint=str,
58
  description="Expanded prompt by the LLM",
59
+ metadata={"mellon":"text"}
60
  ),
61
  OutputParam(
62
  "old_prompt",