Update app.py

app.py

@@ -74,9 +74,9 @@ def process_input_image(image_source, download_masks=False):
     
     mask_zip, output_name = create_mask_zip(predicted_image)
     if download_masks:
-        return "Predicted Masked Image", rgb_image, mask_zip, output_name
+        return rgb_image, mask_zip
     else:
-        return "Predicted Masked Image", rgb_image, None, None
+        return rgb_image, None
 
 
 
@@ -176,7 +176,6 @@ with my_app:
                     download_masks_checkbox
                 ],
                 [
-                    output_label,
                     img_output,
                     masks_zip,
                 ]