Update app.py

app.py

@@ -42,10 +42,13 @@ def infer(img):
 
 
 with gr.Blocks() as demo:
-    im = gr.ImageEditor(
-        type="pil"
-    )
-    submit_btn = gr.Button()
+    with gr.Row():
+        with gr.Column():
+            im = gr.ImageEditor(
+                type="pil"
+            )
+            submit_btn = gr.Button()
+        output = gr.AnnotatedImage()
     submit_btn.click(infer, inputs=im, outputs=gr.AnnotatedImage())
 
 demo.launch(debug=True)