diff --git a/invesalius/segmentation/brain/segment.py b/invesalius/segmentation/brain/segment.py
index 62ba8d8..741d95e 100644
--- a/invesalius/segmentation/brain/segment.py
+++ b/invesalius/segmentation/brain/segment.py
@@ -74,8 +74,6 @@ def brain_segment(image, probability_array, comm_array):
         model = keras.models.model_from_json(json_file.read())
     model.load_weights(str(folder.joinpath("model.h5")))
     model.compile("Adam", "binary_crossentropy")
-    
-    keras.utils.plot_model(model, "model.png", show_shapes=True, show_layer_names=True)
 
     image = imagedata_utils.image_normalize(image, 0.0, 1.0, output_dtype=np.float32)
     sums = np.zeros_like(image)
--
libgit2 0.21.2