Update app.py

app.py

@@ -28,7 +28,7 @@ def predict_segmentation(image):
 
     custom_objects = {'dice_coef': dice_coef}
     with tf.keras.utils.custom_object_scope(custom_objects):
-        model = tf.keras.models.load_model("model100.h5")
+        model = tf.keras.models.load_model("model_best.h5")
 
     # Get the prediction
     prediction = model.predict(X_test)