Update app.py

app.py

@@ -72,11 +72,11 @@ def process_input_image(image_source, download_masks=False):
     predicted_image = predicted_image[0, :, :]
     rgb_image = label_to_rgb(predicted_image)
     
+    mask_zip, output_name = create_mask_zip(predicted_image)
     if download_masks:
-        mask_zip, output_name = create_mask_zip(predicted_image)
         return "Predicted Masked Image", rgb_image, mask_zip, output_name
     else:
-        return "Predicted Masked Image", rgb_image
+        return "Predicted Masked Image", rgb_image, None, None
 
 
 my_app = gr.Blocks()
@@ -150,6 +150,7 @@ with my_app:
                     output_label,
                     img_output,
                     masks_zip,
-                ]
+                ],
+                show_hidden_outputs=True
             )
 my_app.launch(debug=True)