Update app.py

[fix] norm col

app.py

@@ -35,6 +35,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) 
     mask = F.interpolate(torch.from_numpy(mask).unsqueeze(0).unsqueeze(0),size=[img.shape[0],img.shape[1]],mode='bilinear').squeeze().numpy()
     col = torch.from_numpy(mask).squeeze().unsqueeze(2).repeat(1,1,3)
+    col=col/col.max()
     mask_torch=torch.from_numpy(mask).squeeze().unsqueeze(2).repeat(1,1,3)
     maks_torch=mask_torch/mask_torch.max()
     col[:,:,0]=0