Customizable fill

This help use to select what they want to fill as option.

app.py

@@ -44,17 +44,9 @@ pipe = StableDiffusionXLFillPipeline.from_pretrained(
 
 pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
 
-prompt = "high quality"
-(
-    prompt_embeds,
-    negative_prompt_embeds,
-    pooled_prompt_embeds,
-    negative_pooled_prompt_embeds,
-) = pipe.encode_prompt(prompt, "cuda", True)
-
 
 @spaces.GPU
-def fill_image(image, model_selection):
+def fill_image(image, model_selection, prompt_text):
     source = image["background"]
     mask = image["layers"][0]
 
@@ -62,6 +54,14 @@ def fill_image(image, model_selection):
     binary_mask = alpha_channel.point(lambda p: p > 0 and 255)
     cnet_image = source.copy()
     cnet_image.paste(0, (0, 0), binary_mask)
+    
+    prompt = f"high quality, {prompt_text}"
+    (
+        prompt_embeds,
+        negative_prompt_embeds,
+        pooled_prompt_embeds,
+        negative_pooled_prompt_embeds,
+    ) = pipe.encode_prompt(prompt, "cuda", True)
 
     for image in pipe(
         prompt_embeds=prompt_embeds,
@@ -119,15 +119,18 @@ with gr.Blocks(css=css) as demo:
         label="Model",
     )
 
+    with gr.Row():
+        prompt_input = gr.Textbox(label="Prompt (Optional)")
+
     run_button.click(
         fn=clear_result,
         inputs=None,
         outputs=result,
     ).then(
         fn=fill_image,
-        inputs=[input_image, model_selection],
+        inputs=[input_image, model_selection, prompt_input],
         outputs=result,
     )
 
 
-demo.launch(share=False)
+demo.launch(share=False)