Update app.py

app.py

@@ -141,7 +141,7 @@ with image_blocks as demo:
     )
     with gr.Row():
         with gr.Column():
-            image = gr.Image(source='upload', tool='sketch', elem_id="image_upload", type="pil").style(height=400)
+            image = gr.Image(source='upload', tool='sketch', elem_id="image_upload", type="pil", label="Upload").style(height=400)
             with gr.Box(elem_id="mask_radio").style(border=False):
                 radio = gr.Radio(["draw a mask above", "type what to mask below"], value="draw a mask above", show_label=False, interactive=True).style(container=False)
                 word_mask = gr.Textbox(label = "What to find in your image", interactive=False, elem_id="word_mask", placeholder="Disabled").style(container=False)
@@ -155,7 +155,7 @@ with image_blocks as demo:
             }""")
             btn = gr.Button("Run")
         with gr.Column():
-            result = gr.Image()
+            result = gr.Image(label="Result")
         btn.click(fn=predict, inputs=[radio, image, word_mask, prompt], outputs=result)
     gr.HTML(
             """