Hugging Face's logo

Spaces:
ciasimbaya
/
Taller3_Cancer_Ultrasound_Image_Segmentation
Runtime error

App Files Community
ciasimbaya commited on
Commit
365c363
•
1 Parent(s): c45e6d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def build_model(input_shape):
46
 
47
  output_layer = Conv2D(1, 1, padding="same", activation="sigmoid")(d4)
48
  model = Model(input_layer, output_layer, name="U-Net")
49
- model.load_weights('./BreastCancerSegmentation.h5')
50
  return model
51
 
52
  def preprocess_image(image, size=128):
 
46
 
47
  output_layer = Conv2D(1, 1, padding="same", activation="sigmoid")(d4)
48
  model = Model(input_layer, output_layer, name="U-Net")
49
+ model.load_weights('BreastCancerSegmentation.h5')
50
  return model
51
 
52
  def preprocess_image(image, size=128):