Update app.py

app.py

@@ -7,27 +7,6 @@ from tensorflow.keras.layers import Input, Conv2D
 from tensorflow.keras.models import Model
 
 size = 128
-
-def build_model(input_shape):
-    input_layer = Input(input_shape)
-    
-    s1, p1 = encoder_block(input_layer, 64)
-    s2, p2 = encoder_block(p1, 128)
-    s3, p3 = encoder_block(p2, 256)
-    s4, p4 = encoder_block(p3, 512)
-
-    b1 = conv_block(p4, 1024)
-
-    d1 = decoder_block(b1, s4, 512)
-    d2 = decoder_block(d1, s3, 256)
-    d3 = decoder_block(d2, s2, 128)
-    d4 = decoder_block(d3, s1, 64)
-    
-    output_layer = Conv2D(1, 1, padding="same", activation="sigmoid")(d4)
-    
-    model = Model(input_layer, output_layer, name="U-Net")
-    return model
-
 model = build_model(input_shape=(size, size, 1))
 model.load_weights('BreastCancerSegmentation.h5')