Update app.py

app.py

@@ -68,7 +68,7 @@ with gr.Blocks() as demo:
             submit_button_component = gr.Button(
                 value='Submit', variant='primary')
         with gr.Column():
-            image_output_component = gr.Image(label='Output mask')
+            image_output_component = gr.Image(label='Output mask', format='png')
 
     submit_button_component.click(
         fn=process_image,