let's keep Image for now

app.py

@@ -121,7 +121,7 @@ def interface():
         with gr.Row():
             btn1 = gr.Button("Generate")
         with gr.Row():
-            watermarked_image = gr.ImageEditor(type="pil", 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")