Update app.py

app.py

@@ -16,7 +16,12 @@ assert os.path.exists("sd-ggml-cpp-dp")
 '''
 
 os.system("pip install huggingface_hub")
-def make_and_download_clean_dir(repo_name = "svjack/sd-ggml"):
+#### https://huggingface.co/svjack/sd-ggml-cpp-dp/resolve/main/models/Cyberpunk_Anime_Diffusion-ggml-model_q4_0.bin
+def make_and_download_clean_dir(repo_name = "svjack/sd-ggml", 
+                               rp_tgt_tail_dict = {
+                                   "models": "wget https://huggingface.co/{}/resolve/main/{}/{}"
+                               }
+                               ):
     import shutil
     import os
     from tqdm import tqdm
@@ -33,9 +38,22 @@ def make_and_download_clean_dir(repo_name = "svjack/sd-ggml"):
         src = path
         tgt = src.split("/")[-1]
         print("downloading {} to {}".format(src, tgt))
-        fs.download(
-           src, tgt, recursive = True
-        )
+        if tgt not in rp_tgt_tail_dict:
+            fs.download(
+               src, tgt, recursive = True
+            )
+        else:
+            tgt_cmd_format = rp_tgt_tail_dict[tgt]
+            os.mkdir(tgt)
+            os.chdir(tgt)
+            sub_fd_list = fs.ls(src, detail = False)
+            for sub_file in tqdm(sub_fd_list):
+                tgt_cmd = tgt_cmd_format.format(
+                repo_name, tgt, sub_file.split("/")[-1]
+                )
+                print("run {}".format(tgt_cmd))
+                os.system(tgt_cmd)
+            os.chdir("..")
     os.chdir("..")
 make_and_download_clean_dir("svjack/sd-ggml")
 os.chdir("sd-ggml")