Spaces:
Sleeping
Sleeping
File size: 448 Bytes
0799705 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import gdown
import torch
from msanii.config import DemoConfig
from msanii.demo import run_demo
# Download checkpoints
id = "1G9kF0r5vxYXPSdSuv4t3GR-sBO8xGFCe"
output = "msanii.pt"
gdown.download(id=id, output=output, quiet=True)
# Setup app config
config = DemoConfig(
ckpt_path="msanii.pt",
device=("cuda" if torch.cuda.is_available() else "cpu"),
dtype="float32",
launch=False,
)
demo = run_demo(config)
demo.queue().launch()
|