Update app.py

app.py

@@ -43,11 +43,11 @@ def build_model(input_shape):
     d4 = decoder_block(d3, s1, 64)
     
     output_layer = Conv2D(1, 1, padding="same", activation="sigmoid")(d4)
-    model = build_model(input_shape=(size, size, 1))
     model.compile(loss="binary_crossentropy", optimizer="Adam", metrics=["accuracy"])
-    model.load_weights('BreastCancerSegmentation.h5')
-    model = Model(input_layer, output_layer, name="U-Net")
     return model
+    
+model.load_weights('path/to/BreastCancerSegmentation.h5')
+model = build_model(input_shape=(size, size, 1))
 
 def preprocess_image(image, size=128):
     image = cv2.resize(image, (size, size))