Spaces:
Runtime error
Runtime error
File size: 431 Bytes
4a285f6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import os
import gdown
os.makedirs('./saved_models/u2net', exist_ok=True)
os.makedirs('./saved_models/u2net_portrait', exist_ok=True)
gdown.download('https://drive.google.com/uc?id=1ao1ovG1Qtx4b7EoskHXmi2E9rp5CHLcZ',
'./saved_models/u2net/u2net.pth',
quiet=False)
gdown.download('https://drive.google.com/uc?id=1IG3HdpcRiDoWNookbncQjeaPN28t90yW',
'./saved_models/u2net_portrait/u2net_portrait.pth',
quiet=False)
|