Update app.py

app.py

@@ -88,7 +88,7 @@ def infer(edit_images,
           strength=1.0, 
           num_inference_steps=30, 
           true_cfg_scale=4.0,
-          rewrite_prompt=True,
+          prompt_enhance=True,
           progress=gr.Progress(track_tqdm=True)):
     
     image = edit_images["background"]
@@ -97,7 +97,7 @@ def infer(edit_images,
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
-    if rewrite_prompt:
+    if prompt_enhance:
         enhanced_prompt = rewrite_prompt(prompt)
         print(f"Original prompt: {prompt}\nEnhanced prompt: {enhanced_prompt}")
         prompt = enhanced_prompt
@@ -217,7 +217,7 @@ with gr.Blocks(css=css, theme=gr.themes.Citrus()) as demo:
                 value=30,
             )
         
-        rewrite_prompt = gr.Checkbox(
+        prompt_enhance = gr.Checkbox(
             label="Enhance prompt (using HF Inference)", 
             value=True
         )
@@ -239,7 +239,7 @@ with gr.Blocks(css=css, theme=gr.themes.Citrus()) as demo:
         show_api=False
     ).then(
         fn=infer,
-        inputs=[edit_image, prompt, negative_prompt, seed, randomize_seed, strength, num_inference_steps, true_cfg_scale, rewrite_prompt],
+        inputs=[edit_image, prompt, negative_prompt, seed, randomize_seed, strength, num_inference_steps, true_cfg_scale, prompt_enhance],
         outputs=[result, seed]
     ).then(
         fn=lambda: gr.update(visible=True),