Update app.py

app.py

@@ -74,11 +74,9 @@ def process_input_image(image_source, download_masks=False):
     
     mask_zip, output_name = create_mask_zip(predicted_image)
     if download_masks:
-        return rgb_image, mask_zip
+        return "Predicted Masked Image", rgb_image, mask_zip.getvalue(), output_name
     else:
-        return rgb_image, None
-
-
+        return "Predicted Masked Image", rgb_image, None, None
 
 
 my_app = gr.Blocks()