soiz1 commited on
Commit
292d826
·
verified ·
1 Parent(s): c0335a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import numpy as np
6
  import torch
7
  from PIL import Image
8
  from scipy.ndimage import binary_dilation, binary_erosion
 
9
 
10
  # FAQ:
11
  # Why is the detection model needed since the meaningful_diff can already detect watermark masks?
 
6
  import torch
7
  from PIL import Image
8
  from scipy.ndimage import binary_dilation, binary_erosion
9
+ torch.serialization.add_safe_globals([object])
10
 
11
  # FAQ:
12
  # Why is the detection model needed since the meaningful_diff can already detect watermark masks?