Add application

app.py

@@ -0,0 +1,20 @@
+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()

requirements.txt

@@ -0,0 +1,2 @@
+gdown==4.6.0
+-e git+https://github.com/Kinyugo/msanii.git#egg=msanii