ironjr commited on
Commit
749679c
·
verified ·
1 Parent(s): 855f3df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -564,7 +564,7 @@ def draw(state, drawpad):
564
  # mask_stds = [0] + [state.mask_stds[v] for v in has_masks]
565
 
566
  for i in range(len(has_masks)):
567
- model.update_single_layer(
568
  idx=i,
569
  mask=masks[i:i + 1],
570
  mask_strength=mask_strengths[i],
 
564
  # mask_stds = [0] + [state.mask_stds[v] for v in has_masks]
565
 
566
  for i in range(len(has_masks)):
567
+ model.update(
568
  idx=i,
569
  mask=masks[i:i + 1],
570
  mask_strength=mask_strengths[i],