fix row

app.py

@@ -29,12 +29,18 @@ interface = Interface(pre_pipe=preprocessing,
                       seg_pipe=seg_net)
 
 
+def generate_trimap(original, mask):
+    trimap(original_image=original, mask=mask)
+
+
 def predict(image):
     return interface([image])[0]
 
 
 footer = r"""
 <center>
+<img src='https://raw.githubusercontent.com/leonelhs/image-background-remove-tool/master/docs/imgs/logo.png' alt='CarveKit' width="200" height="80">
+</br>
 <b>
 Demo based on <a href='https://github.com/OPHoperHPO/image-background-remove-tool'>CarveKit</a>
 </b>
@@ -42,10 +48,11 @@ Demo based on <a href='https://github.com/OPHoperHPO/image-background-remove-too
 """
 
 with gr.Blocks(title="CarveKit") as app:
-    gr.HTML("<center><h1><b>CarveKit</b></h1></center>")
-    gr.HTML("<center><h3>Automated high-quality background removal framework for an image using neural networks.</h3></center>")
-    with gr.Row():
-        with gr.Row().style(equal_height=False):
+    gr.Markdown("<center><h1><b>CarveKit</b></h1></center>")
+    gr.HTML(
+        "<center><h3>Automated high-quality background removal framework for an image using neural networks.</h3></center>")
+    with gr.Row().style(equal_height=False):
+        with gr.Column():
             input_img = gr.Image(type="pil", label="Input image")
             run_btn = gr.Button(variant="primary")
         with gr.Column():