Adding missed mask output

app.py

@@ -197,7 +197,7 @@ with gr.Blocks() as demo:
         satellite_output = gr.Image(label="Generated Satellite Image")
         mask_output = gr.Image(label="Mask")
         image_prompt = gr.Textbox(label="Image Prompt Used")
-    submit_btn.click(handle_query, inputs=[query_input], outputs=[map_output, satellite_output, image_prompt])
+    submit_btn.click(handle_query, inputs=[query_input], outputs=[map_output, satellite_output, mask_output, image_prompt])
 
 if __name__ == "__main__":
     demo.launch()