Update app.py

flip + shift ops

app.py

@@ -77,9 +77,11 @@ def edit(input_image,
                     warm_up=1,
                     neg_guidance=False,
                     flip=False,
-                    h_shift=0,
-                    v_shift=0):
-    left,right, top, bottom=0,200,0,0
+                    left = 0,
+                    right = 0,
+                    top = 0,
+                    bottom = 0):
+
     # offsets=(0,0,0,0)
     x0 = load_512(input_image, left,right, top, bottom, device)
 
@@ -168,8 +170,10 @@ with gr.Blocks() as demo:
             skip = gr.Slider(minimum=0, maximum=40, value=36, precision=0, label="Skip Steps", interactive=True)
             tar_cfg_scale = gr.Slider(minimum=7, maximum=18,value=15, label=f"Guidance Scale", interactive=True)
             flip = gr.Checkbox(label="Flip")
-            h_shift = gr.Number(value=0, precision=0, label="Horizontal Shift", interactive=True)
-            v_shift =  gr.Number(value=0, precision=0, label="Vertical Shift", interactive=True)
+            left = gr.Number(value=0, precision=0, label="Left Shift", interactive=True)
+            right = gr.Number(value=0, precision=0, label="Right Shift", interactive=True)
+            top = gr.Number(value=0, precision=0, label="Top Shift", interactive=True)
+            bottom = gr.Number(value=0, precision=0, label="Bottom Shift", interactive=True)
             # edit
             sega_edit_guidance = gr.Slider(value=10, label=f"SEGA Edit Guidance Scale", interactive=True)
             # warm_up = gr.Number(value=1, label=f"SEGA Warm-up Steps", interactive=True)
@@ -192,8 +196,10 @@ with gr.Blocks() as demo:
                     # warm_up,
                     # neg_guidance,
                     flip,
-                    h_shift,
-                    v_shift
+                    left,
+                    right,
+                    top,
+                    bottom
         ],
         outputs=[ddpm_edited_image, sega_edited_image],
     )