Update app.py

app.py

@@ -2,16 +2,19 @@ import gradio as gr
 import imageio
 
 def dummy(img):
-  imageio.imwrite("output_image.png", img["mask"])
-  return img["image"], img["mask"]
+    imageio.imwrite("output_image.png", img["mask"])
+    return img["image"], img["mask"]
 
 with gr.Blocks() as demo:
-  with gr.Row():
-    img = gr.Image(tool="sketch", label="base image", show_label=True)
-  with gr.Row():
-    img1 = gr.Image()
-    img2 = gr.Image(label="mask image", show_label=True)
-  btn = gr.Button()
-  btn.click(dummy, img, [img1, img2])
+    with gr.Row():
+        img = gr.Image(
+            tool="sketch", label="base image", show_label=True,
+            style={"height": "500px", "max-width": "100%", "@media (max-width: 768px)": {"height": "300px"}}
+        )
+    with gr.Row():
+        img1 = gr.Image()
+        img2 = gr.Image(label="mask image", show_label=True)
+    btn = gr.Button()
+    btn.click(dummy, img, [img1, img2])
 
-demo.launch(debug=True)
+demo.launch(debug=True)