Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from PIL import Image
 
+size = 128, size==size
+
 def conv_block(input, num_filters):
     conv = Conv2D(num_filters, (3, 3), activation="relu", padding="same", kernel_initializer='he_normal')(input)
     conv = Conv2D(num_filters, (3, 3), activation="relu", padding="same", kernel_initializer='he_normal')(conv)
@@ -36,12 +38,9 @@ def build_model(input_shape):
     model = build_model(input_shape=(size, size, 1))
     model.compile(loss="binary_crossentropy", optimizer="Adam", metrics=["accuracy"])
     model.load_weights('BreastCancerSegmentation.h5')
-    size = 128, size==size
     model = Model(input_layer, output_layer, name="U-Net")
     return model
 
-
-
 def preprocess_image(image, size=128):
     image = cv2.resize(image, (size, size))
     image = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)