Update app.py

app.py

@@ -563,13 +563,18 @@ def draw(state, drawpad):
     #     mask_strengths = [1] + [state.mask_strengths[v] for v in has_masks]
     #     mask_stds = [0] + [state.mask_stds[v] for v in has_masks]
 
-    for i in range(len(has_masks)):
-        model.update(
-            idx=i,
-            mask=masks[i:i + 1],
-            mask_strength=mask_strengths[i],
-            mask_std=mask_stds[i],
-        )
+    model.update_masks(
+        mask=masks,
+        mask_strength=mask_strengths,
+        mask_std=mask_stds,
+    )
+    # for i in range(len(has_masks)):
+    #     model.update(
+    #         idx=i,
+    #         mask=masks[i:i + 1],
+    #         mask_strength=mask_strengths[i],
+    #         mask_std=mask_stds[i],
+    #     )
     # data = dict(
     #     masks=masks,
     #     mask_strengths=mask_strengths,