Spaces:

hasibzunair
/

fifa-tryon-demo

Runtime error

App Files Files Community

fifa-tryon-demo / setup_model_weights.py

hasibzunair's picture

added files

4a285f6 over 1 year ago

raw history blame contribute delete

No virus

431 Bytes

	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)