Update app.py

app.py

@@ -43,11 +43,13 @@ 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))
@@ -73,4 +75,4 @@ if __name__ == "__main__":
         outputs=gr.Image(type="pil", label="Breast Cancer Mask"),
         title="Breast Cancer Ultrasound Image Segmentation",
         description="Check out this exciting development in the field of breast cancer diagnosis and treatment! A demo of Breast Cancer Ultrasound Image Segmentation has been developed. Upload an image file, or try out one of the examples below!"
-    ).launch(debug=True)
+    ).launch(debug=True)