Update app.py

app.py

@@ -564,7 +564,7 @@ def draw(state, drawpad):
     #     mask_stds = [0] + [state.mask_stds[v] for v in has_masks]
 
     for i in range(len(has_masks)):
-        model.update_single_layer(
+        model.update(
             idx=i,
             mask=masks[i:i + 1],
             mask_strength=mask_strengths[i],