Update app.py

app.py

@@ -46,7 +46,12 @@ def build_model(input_shape):
     
     output_layer = Conv2D(1, 1, padding="same", activation="sigmoid")(d4)                                                               
     model = Model(input_layer, output_layer, name="U-Net")
-    model.load_weights('BreastCancerSegmentation.h5')
+    try:
+        model.load_weights('./BreastCancerSegmentation.h5')
+    except Exception as e:
+        print("Error al cargar los pesos del modelo:")
+        print(e)
+        exit(1)
     return model
     
 def preprocess_image(image, size=128):