let's try this

app.py

@@ -121,14 +121,14 @@ def interface():
         with gr.Row():
             btn1 = gr.Button("Generate")
         with gr.Row():
-            watermarked_image = gr.Image(type="pil")#.style(width=512, height=512)
+            watermarked_image = gr.Image(type="pil", width=512, height=512)
             with gr.Column():
                 downscale = gr.Slider(1, 3, value=1, step=0.1, label="Downscale ratio")
                 saturation = gr.Slider(0, 2, value=1, step=0.1, label="Color saturation")
                 jpeg_compression = gr.Slider(value=100, step=5, label="JPEG quality")
                 btn2 = gr.Button("Attack & Detect")
                 with gr.Row():
-                    attacked_image = gr.Image(type="pil", tool="select")#.style(width=256)
+                    attacked_image = gr.Image(type="pil", width=256)
                     detection_label = gr.Label(label="Detection info")
         btn1.click(fn=backend.generate, inputs=[mode, seed, inp], outputs=[watermarked_image], api_name="generate")
         btn2.click(fn=backend.attack_detect, inputs=[watermarked_image, jpeg_compression, downscale, saturation], outputs=[attacked_image, detection_label], api_name="detect")