Update app.py

app.py

@@ -42,11 +42,9 @@ def build_model(input_shape):
     d3 = decoder_block(d2, s2, 128)
     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')
+    output_layer = Conv2D(1, 1, padding="same", activation="sigmoid")(d4)                                                               
     model = Model(input_layer, output_layer, name="U-Net")
+    model.load_weights('BreastCancerSegmentation.h5')
     return model
     
 def preprocess_image(image, size=128):
@@ -67,6 +65,7 @@ def segment(image):
     return mask_image
 
 if __name__ == "__main__":
+    model = build_model(input_shape=(size, size, 1))
     gr.Interface(
         fn=segment,
         inputs="image",