Update app.py

app.py

@@ -1,16 +1,13 @@
 import gradio as gr
 from PIL import Image
-import tensorflow as tf
-from keras.models import Model
-from keras.layers import Input, Conv2D, MaxPooling2D, Conv2DTranspose, concatenate
-from keras.optimizers import Adam
+
+size=128
+model.load_weights('BreastCancerSegmentation.h5')
 
 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))
     image = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)