model.py

import tensorflow as tf

def create_simplified_model():
    """Creates a very basic segmentation model."""
    model = tf.keras.Sequential([
        tf.keras.layers.Conv2D(32, 3, activation='relu', padding='same', input_shape=(256, 256, 1)),
        tf.keras.layers.Conv2D(1, 1, activation='sigmoid')  # output a mask
    ])
    model.compile(optimizer='adam', loss='binary_crossentropy')
    return model