Omnibus commited on
Commit
29136da
·
1 Parent(s): b2358ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -11,5 +11,13 @@ with gr.Blocks() as app:
11
  msg_box = gr.Textbox(label="System Message")
12
  with gr.Column():
13
  out_im = gr.Image(label="Watermarked Image")
 
 
 
 
 
 
 
14
  mark_btn.click(utils.encode,[inp_im,inp_mark],[out_im,msg_box])
 
15
  app.launch()
 
11
  msg_box = gr.Textbox(label="System Message")
12
  with gr.Column():
13
  out_im = gr.Image(label="Watermarked Image")
14
+ with gr.Tab("Detect Watermark"):
15
+ with gr.Row():
16
+ with gr.Column():
17
+ det_im = gr.Image(label="Watermarked Image",type="filepath")
18
+ det_btn = gr.Button("Detect")
19
+ with gr.Column():
20
+ det_msg = gr.Textbox(label="Detected Watermark",lines=6, max_lines=50)
21
  mark_btn.click(utils.encode,[inp_im,inp_mark],[out_im,msg_box])
22
+ det_btn.click(utils.decode,[det_im],det_msg)
23
  app.launch()