Update app.py

app.py

@@ -407,15 +407,16 @@ def run(state, drawpad):
     tic = time.time()
     while True:
         # Receive real-time mask inputs from the main process.
-        msg = conn.recv()
-        print(msg + '  Received!!!')
-        # for i in range(opt.max_palettes):
-        #     model.update_single_layer(
-        #         idx=i,
-        #         mask=masks[i],
-        #         mask_strength=mask_strengths[i],
-        #         mask_std=mask_stds[i],
-        #     )
+        data = conn.recv()
+        if data is not None:
+            print('Received data!!!')
+            for i in range(opt.max_palettes):
+                model.update_single_layer(
+                    idx=i,
+                    mask=data['masks'][i],
+                    mask_strength=data['mask_strengths'][i],
+                    mask_std=data['mask_stds'][i],
+                )
 
         yield [state, model()]
         toc = time.time()
@@ -472,7 +473,12 @@ def draw(state, drawpad):
     #         mask_strength=mask_strengths[i],
     #         mask_std=mask_stds[i],
     #     )
-    conn.send('Hello!!!!')
+    data = dict(
+        masks=masks,
+        mask_strengths=mask_strengths,
+        mask_stds=mask_stds,
+    )
+    conn.send(data)
     conn.close()
 
 ### Load examples