Update app.py

app.py

@@ -26,6 +26,7 @@ def real_dice_coeff(y_true, y_pred):
     dice_score = (2.0 * intersection + smooth) / (K.sum(y_true_flatten) + K.sum(y_pred_flatten) + smooth)
     return dice_score
 
+weights = [0.5,0.5] # hyper parameter
 
 dice_loss = sm.losses.DiceLoss(class_weights = weights)
 focal_loss = sm.losses.CategoricalFocalLoss()