Spaces:
Running
Running
Update app.py (#1)
Browse files- Update app.py (292d826aa8da4658bc6263078ad797d42f8227bc)
Co-authored-by: izum00 <[email protected]>
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?
|