Update app.py

app.py

@@ -273,7 +273,7 @@ def run_nnunet_predict(nifti_file,hd_bet=False):
             nib.save(nib.Nifti1Image(labeled_mask.astype(np.int16), img.affine), labeled_mask_path)
 
             # Return paths for the Gradio interface
-            return new_output_file, input_slice_path, output_slice_path
+            return new_output_file, input_slice_path, output_slice_path, labeled_mask_path
         else:
             return "Error: Output file not found."
     except subprocess.CalledProcessError as e: