fix ckpt downloading

app.py

@@ -1,10 +1,20 @@
 import subprocess as sp
-import sys
+import os
+
+# Download if not exists
+os.makedirs("checkpoints", exist_ok=True)
+
+if not os.path.exists("checkpoints/text2semantic-medium-v1-2k.pth"):
+    print("Downloading text2semantic-medium-v1-2k.pth")
+    sp.run(["wget", "-O", "checkpoints/text2semantic-medium-v1-2k.pth", os.environ["CKPT_SEMANTIC"]])
 
-sp.check_call(["bash", "setup.sh"], shell=True, stdout=sys.stdout, stderr=sys.stderr)
+if not os.path.exists("checkpoints/vq-gan-group-fsq-2x1024.pth"):
+    print("Downloading vq-gan-group-fsq-2x1024.pth")
+    sp.run(["wget", "-O", "checkpoints/vq-gan-group-fsq-2x1024.pth", os.environ["CKPT_VQGAN"]])
+
+print("All checkpoints downloaded")
 
 import html
-import os
 from argparse import ArgumentParser
 from io import BytesIO
 from pathlib import Path

setup.sh

@@ -1,18 +0,0 @@
-#!/bin/bash
-set -e
-
-mkdir -p checkpoints
-
-if [ -e checkpoints/text2semantic-medium-v1-2k.pth ]; then
-    echo "checkpoints/text2semantic-medium-v1-2k.pth already exists"
-else
-    echo "Downloading text2semantic-medium-v1-2k.pth"
-    wget -O checkpoints/text2semantic-medium-v1-2k.pth $CKPT_SEMANTIC
-fi
-
-if [ -e checkpoints/vq-gan-group-fsq-2x1024.pth ]; then
-    echo "checkpoints/vq-gan-group-fsq-2x1024.pth already exists"
-else
-    echo "Downloading vq-gan-group-fsq-2x1024.pth"
-    wget -O checkpoints/vq-gan-group-fsq-2x1024.pth $CKPT_VQGAN
-fi