Update app.py

[fix] col out of index

app.py

@@ -34,7 +34,7 @@ def generate_mask(img, coord):
     #
     mask = sepia((img*0.999999).astype(np.uint8),(img*0.999999).astype(np.uint8),(img*0.999999).astype(np.uint8),(img*0.999999).astype(np.uint8),(img*0.999999).astype(np.uint8), stack_image=False) 
     col = torch.from_numpy(mask).squeeze()
-    mask_torch=torch.from_numpy(mask).squeeze()
+    mask_torch=torch.from_numpy(mask).squeeze().unsqueeze(2).repeat(1,1,3)
     maks_torch=mask_torch/mask_torch.max()
     col[:,:,0]=0
     mix = (1-mask_torch)*img+mask_torch*img*0.7+mask_torch*col*0.3