Update app.py

app.py

@@ -5,10 +5,10 @@ from keras.layers import Input, Conv2D, MaxPooling2D, Conv2DTranspose, concatena
 from keras.optimizers import Adam
 
 def build_model(input_shape):
-size=128
-model = build_model(input_shape=(size, size, 1))
-model.load_weights('BreastCancerSegmentation.h5')
- return model
+    size=128
+    model = build_model(input_shape=(size, size, 1))
+    model.load_weights('BreastCancerSegmentation.h5')
+    return model
 
 def preprocess_image(image, size: int=128):
     image = cv2.resize(image, (size,size))