Update app.py

app.py

@@ -74,11 +74,14 @@ def process_input_image(image_source, download_masks=False):
     
     mask_zip, output_name = create_mask_zip(predicted_image)
     if download_masks:
+        masks_zip.set_attributes(label="Download Masks", filename="masks.zip", hidden=False)
         return "Predicted Masked Image", rgb_image, mask_zip, output_name
     else:
+        masks_zip.set_attributes(hidden=True)
         return "Predicted Masked Image", rgb_image, None, None
 
 
+
 my_app = gr.Blocks()
 
 
@@ -150,7 +153,6 @@ with my_app:
                     output_label,
                     img_output,
                     masks_zip,
-                ],
-                show_hidden_outputs=True
+                ]
             )
 my_app.launch(debug=True)