Update app.py

app.py

@@ -180,6 +180,7 @@ def detect_watermark(det_im):
 # Gradio app
 with gr.Blocks() as app:
     with gr.Tab("Add Watermark"):
+        cho = gr.Radio(choices=["stegan", ""] ,value="stenography", label="Watermark Method")
         with gr.Row():
             with gr.Column():
                 inp_im = gr.Image(label="Input Image", type="filepath")
@@ -190,7 +191,7 @@ with gr.Blocks() as app:
             with gr.Column():
                 out_im = gr.Image(label="Watermarked Image")
 
-        mark_btn.click(choose_encode, inputs=[inp_im, inp_mark], outputs=[out_im, msg_box, file_output])
+        mark_btn.click(choose_encode, inputs=[inp_im, inp_mark, cho], outputs=[out_im, msg_box, file_output])
 
     with gr.Tab("Detect Watermark"):
         with gr.Row():