app.py

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()