chore: update layout

SSL_watermark.py

@@ -84,4 +84,4 @@ def decode(image):
     loss_R = -(rho * dot_product**2 - norm**2) # B-B -> B
 
     text_marked = "marked" if loss_R < 0 else "unmarked"
-    return f'Image is {text_marked}, with p-value={10**log10_pvalue}'
+    return f'Image is {text_marked}'

app.py

@@ -81,7 +81,7 @@ with gr.Blocks() as demo:
 
             with gr.Column():
                 gr.Markdown("### Applied watermark image")
-                output_image = gr.Image(show_label=False)
+                output_image = gr.Image(show_label=False).style(height=500)
         with gr.Row():
             apply_button =gr.Button("Apply")
         
@@ -97,7 +97,7 @@ with gr.Blocks() as demo:
                 )
             with gr.Column():
                 gr.Markdown("### Extracted watermark")
-                extracted_watermark = gr.Image(type='pil')
+                extracted_watermark = gr.Image(type='pil').style(height=500)
         extract_button = gr.Button("Extract")
     
     radio_button.change(