Update app.py

app.py

@@ -16,28 +16,29 @@ footer {
 """
 
 with gr.Blocks(css=css) as app:
-
+    gr.Markdown("# Image Watermarking Tool")
     gr.Markdown("[![Visitors](https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Ffantos-watermark.hf.space&countColor=%23263759)](https://visitorbadge.io/status?path=https%3A%2F%2Ffantos-watermark.hf.space)")
-
     
     with gr.Tab("Add Watermark"):
-        cho = gr.Radio(choices=["stegan","pnginfo"],value="stegan")
+        cho = gr.Radio(choices=["stegan", "pnginfo"], value="stegan")
         with gr.Row():
-
             with gr.Column():
-                inp_im = gr.Image(label="Input Image",type="filepath")
+                inp_im = gr.Image(label="Input Image", type="filepath")
                 inp_mark = gr.Textbox(label="Watermark")
-                mark_btn = gr.Button()
+                mark_btn = gr.Button("Add Watermark")
                 msg_box = gr.Textbox(label="System Message")
             with gr.Column():
                 out_im = gr.Image(label="Watermarked Image")
+    
     with gr.Tab("Detect Watermark"):
         with gr.Row():
             with gr.Column():
-                det_im = gr.Image(label="Watermarked Image",type="filepath")
+                det_im = gr.Image(label="Watermarked Image", type="filepath")
                 det_btn = gr.Button("Detect")
             with gr.Column():
-                det_msg = gr.Textbox(label="Detected Watermark",lines=6, max_lines=50)
-    mark_btn.click(choose_encode,[inp_im,inp_mark,cho],[out_im,msg_box])
-    det_btn.click(utils.decode,[det_im],det_msg)
+                det_msg = gr.Textbox(label="Detected Watermark", lines=6, max_lines=50)
+    
+    mark_btn.click(choose_encode, [inp_im, inp_mark, cho], [out_im, msg_box])
+    det_btn.click(utils.decode, [det_im], det_msg)
+
 app.launch()