add deltaedit

app.py

@@ -110,8 +110,13 @@ def edit_image(orig_image, edit_direction, edit_power, align, mask, progress=gr.
     return gr.update(value=output_edited, visible=True), gr.update(value=output_inv, visible=True), gr.update(visible=False)
 
 
-def edit_image_clip(orig_image, neutral_prompt, target_prompt, disentanglement, edit_power, align, mask, progress=gr.Progress(track_tqdm=True)):
-    edit_direction = "_".join(["styleclip_global", neutral_prompt, target_prompt, str(disentanglement)])
+def edit_image_clip(orig_image, neutral_prompt, target_prompt, disentanglement, edit_power, align, mask, edit_method, progress=gr.Progress(track_tqdm=True)):
+    if edit_method == "StyleClip":
+        edit_direction = "_".join(["styleclip_global", neutral_prompt, target_prompt, str(disentanglement)])
+    else:
+        edit_power = edit_power / 10
+        disentanglement = disentanglement / 3
+        edit_direction = "_".join(["deltaedit", neutral_prompt, target_prompt, str(disentanglement)])
     return edit_image(orig_image, edit_direction, edit_power, align, mask, progress=None)
 
 
@@ -177,9 +182,10 @@ def get_demo():
                         predef_editing_power = gr.Slider(-20, 20, value=7, step=0.1, label="Editing power")
                     btn_predef = gr.Button("Edit image")
 
-                with gr.Accordion("Text Prompt (StyleClip) Editings", open=False):
+                with gr.Accordion("Text Prompt Editings", open=False):
                     with gr.Accordion("Description", open=False):
-                        gr.Markdown('''You can alse use editings from text prompts via **StyleClip Global Mapper** (https://arxiv.org/abs/2103.17249). You just need to choose:
+                        gr.Markdown('''You can alse use editings from text prompts via **StyleClip Global Mapper** (https://arxiv.org/abs/2103.17249) or **DeltaEdit** (https://arxiv.org/abs/2303.06285). You just need to choose:
+                        **Method** -- method to use, StyleClip or DeltaEdit
 
                         **Editing power** -- the greater the absolute value of this parameter, the more the selected edit will appear.
 
@@ -189,7 +195,7 @@ def get_demo():
 
                         **Disentanglement** -- positive number, the less this attribute -- the more related attributes will also be changed (e.g. for grey hair editing, wrinkle, skin colour and glasses may also be edited)
                           ''')
-
+                    edit_method = gr.Dropdown(["StyleClip", "DeltaEdit"], label="Editing method", value="StyleClip")
                     neutral_prompt = gr.Textbox(value="face with hair", label="Neutreal prompt (e.g. 'a face')")
                     target_prompt = gr.Textbox(value="face with fire hair", label="Target prompt (e.g. 'a smilling face')")
                     styleclip_editing_power = gr.Slider(-50, 50, value=10, step=1, label="Editing power")
@@ -262,7 +268,7 @@ def get_demo():
         )
         btn_clip.click(
             fn=edit_image_clip, 
-            inputs=[input_image, neutral_prompt, target_prompt, disentanglement, styleclip_editing_power, align, mask],
+            inputs=[input_image, neutral_prompt, target_prompt, disentanglement, styleclip_editing_power, align, mask, edit_method],
             outputs=[output_edit, output_inv, error_message]
         )
         btn_mask.click(
@@ -288,4 +294,4 @@ def get_demo():
 
 if __name__ == "__main__":
     demo = get_demo()
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="127.0.0.1", server_port=7860)