Update app.py

app.py

@@ -228,6 +228,12 @@ with gr.Blocks(css=css) as demo:
                     placeholder="describe the edit instruction",
                     container=False,
                 )
+                rewriten_prompt = gr.Text(
+                    label="Rewritten prompt",
+                    show_label=False,
+                    placeholder="Rewritten prompt",
+                    container=False,
+                )
                 run_button = gr.Button("Edit!", variant="primary")
 
             result = gr.Image(label="Result", show_label=False, type="pil")
@@ -272,10 +278,17 @@ with gr.Blocks(css=css) as demo:
 
     gr.on(
         triggers=[run_button.click, prompt.submit],
+        fn=polish_prompt,
+        inputs=[
+            prompt,
+            input_image,
+        ],
+        outputs=[rewriten_prompt],
+    ).then(
         fn=infer,
         inputs=[
             input_image,
-            prompt,
+            rewriten_prompt,
             # negative_prompt is no longer an input from the UI
             seed,
             randomize_seed,
@@ -287,4 +300,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()