Update app.py

app.py

@@ -1,9 +1,14 @@
 import gradio as gr
 from PIL import Image
+from keras.models import Model
+from keras.layers import Input, Conv2D, MaxPooling2D, Conv2DTranspose, concatenate
+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
 
 def preprocess_image(image, size: int=128):
     image = cv2.resize(image, (size,size))