bug fixes

app.py

@@ -78,7 +78,6 @@ def main():
         guidance_scale: float,
         local_edit_word: str,
         invert_local_edit: bool,
-        negative_prompt: str,
         self_replace_steps: float,
         conv_replace_steps: float,
         cross_replace_steps: float,
@@ -98,13 +97,14 @@ def main():
                                   # steps and image size is decided here because edit must be same size as inversion
                                   width=width, height=height)
 
+        invert_local_edit = True if invert_local_edit == "Inverted" else False
         edited_imgs = editpipeline.run_edit(edit_prompt,
                                      guidance_scale=guidance_scale,
                                      local_edit_word=local_edit_word,
                                      # str OPTIONAL # word describing region of image. i.e. an image of a man in a house.
                                      # if you put "man" only the man will be edited, the same word must appear in the inversion prompt as well to work
                                      invert_local_edit=invert_local_edit,  # if True will edit all areas except region
-                                     neg_prompt=negative_prompt,
+                                     neg_prompt="bad quality, low resolution",
                                      cross_replace_steps=cross_replace_steps,  # 0.0 - 0.5 is good
                                      self_replace_steps=self_replace_steps,  # 0.25-0.65
                                      conv_replace_steps=conv_replace_steps, )  # 0.25-0.6, typically like this one lower than self replace)
@@ -128,6 +128,7 @@ def main():
         with gr.Row():
             with gr.Column(scale=3):
                 edit_prompt = gr.Textbox(lines=1, label="Edited version of prompt", interactive=True)
+
         with gr.Row():
             with gr.Column(scale=1, min_width=100):
                 generate_button = gr.Button("Generate")
@@ -147,14 +148,19 @@ def main():
             guidance_scale = gr.Number(value=1.5, label=f"Guidance Scale", interactive=True)
 
         with gr.Row():
-            self_replace_steps = gr.Number(value=50, precision=0, label="Steps", interactive=True)
-            conv_replace_steps = gr.Number(value=1371, precision=0, label="Seed", interactive=True)
-            cross_replace_steps = gr.Textbox(lines=1, label="Local edit word", interactive=True)
+            self_replace_steps = gr.Number(value=50, precision=0, label="Self attn replace steps", interactive=True)
+            conv_replace_steps = gr.Number(value=1371, precision=0, label="Conv replace steps", interactive=True)
+            cross_replace_steps = gr.Textbox(lines=1, label="Cross attn replace steps", interactive=True)
             local_edit_word = gr.Textbox(lines=1, label="Local edit word", interactive=True)
+            invert_local_edit = gr.Radio(
+                ["Normal", "Inverted"],
+                value="Normal",
+                type="index",
+                show_label=False,
+                interactive=True,
+            )
 
         gr.Markdown(help_text)
-        invert_local_edit = True if local_edit_word.value != "" else False
-        negative_prompt = "bad quality, low resolution"
 
         generate_button.click(
             fn=generate,
@@ -165,7 +171,6 @@ def main():
                 guidance_scale,
                 local_edit_word,
                 invert_local_edit,
-                negative_prompt,
                 self_replace_steps,
                 conv_replace_steps,
                 cross_replace_steps,