Update app.py

app.py

@@ -1,13 +1,15 @@
 import gradio as gr
 from PIL import Image
 
-size=128
-model.load_weights('BreastCancerSegmentation.h5')
+size = 128
 
 def build_model(input_shape):
     model = build_model(input_shape=(size, size, 1))
     return model
-    
+
+model = build_model(input_shape=(size, size, 1))
+model.load_weights('BreastCancerSegmentation.h5')
+
 def preprocess_image(image, size: int=128):
     image = cv2.resize(image, (size,size))
     image = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)