Update app.py

app.py

@@ -6,13 +6,40 @@ import numpy as np
 import random
 import shutil
 
+'''
 if not os.path.exists("sd-ggml-cpp-dp"):
     os.system("git clone https://huggingface.co/svjack/sd-ggml-cpp-dp")
 else:
     shutil.rmtree("sd-ggml-cpp-dp")
     os.system("git clone https://huggingface.co/svjack/sd-ggml-cpp-dp")
 assert os.path.exists("sd-ggml-cpp-dp")
-os.chdir("sd-ggml-cpp-dp")
+'''
+
+os.system("pip install huggingface_hub")
+def make_and_download_clean_dir(repo_name = "svjack/sd-ggml"):
+    import shutil
+    import os
+    from tqdm import tqdm
+    from huggingface_hub import HfFileSystem
+    fs = HfFileSystem()
+    req_dir = repo_name.split("/")[-1]
+    if os.path.exists(req_dir):
+        shutil.rmtree(req_dir)
+    os.mkdir(req_dir)
+    os.chdir(req_dir)
+    fd_list = fs.ls(repo_name, detail = False)
+    fd_clean_list = list(filter(lambda x: not x.split("/")[-1].startswith("."), fd_list))
+    for path in tqdm(fd_clean_list):
+        src = path
+        tgt = src.split("/")[-1]
+        print("downloading {} to {}".format(src, tgt))
+        fs.download(
+           src, tgt, recursive = True
+        )
+    os.chdir("..")
+make_and_download_clean_dir("svjack/sd-ggml")
+os.chdir("sd-ggml")
+
 assert os.path.exists("stable-diffusion.cpp")
 os.system("cmake stable-diffusion.cpp")
 os.system("cmake --build . --config Release")