Upload app.py with huggingface_hub

app.py

@@ -22,14 +22,24 @@ from huggingface_hub import snapshot_download
 root = pathlib.Path(__file__).parent
 example_root = os.path.join(root, 'examples')
 ckpt_root = os.path.join(root, 'stablediffusion')
-if not CHECKPOINTS_PATH.exists():
-    snapshot_download(repo_id="ironjr/LucidDreamerDemo", repo_type="model", local_dir=example_root)
-    snapshot_download(repo_id="ironjr/BlazingDriveV11m", repo_type="model", local_dir=os.path.join(ckpt_root, 'Blazing Drive V11m'))
-    snapshot_download(repo_id="ironjr/RealCartoon-PixarV5", repo_type="model", local_dir=os.path.join(ckpt_root, 'RealCartoon-Pixar V5'))
-    snapshot_download(repo_id="ironjr/RealisticVisionV5-1", repo_type="model", local_dir=os.path.join(ckpt_root, 'Realistic Vision V5.1'))
-    snapshot_download(repo_id="runwayml/stable-diffusion-inpainting", repo_type="model", local_dir=os.path.join(ckpt_root, 'SD1-5'))
 
-try:
+d = example_root
+if not os.path.exists(d):
+    snapshot_download(repo_id="ironjr/LucidDreamerDemo", repo_type="model", local_dir=d)
+d = os.path.join(ckpt_root, 'Blazing Drive V11m')
+if not os.path.exists(d):
+    snapshot_download(repo_id="ironjr/BlazingDriveV11m", repo_type="model", local_dir=d)
+d = os.path.join(ckpt_root, 'RealCartoon-Pixar V5')
+if not os.path.exists(d):
+    snapshot_download(repo_id="ironjr/RealCartoon-PixarV5", repo_type="model", local_dir=d)
+d = os.path.join(ckpt_root, 'Realistic Vision V5.1')
+if not os.path.exists(d):
+    snapshot_download(repo_id="ironjr/RealisticVisionV5-1", repo_type="model", local_dir=d)
+d = os.path.join(ckpt_root, 'SD1-5')
+if not os.path.exists(d):
+    snapshot_download(repo_id="runwayml/stable-diffusion-inpainting", repo_type="model", local_dir=d)
+
+try
     import simple_knn
 except ModuleNotFoundError:
     subprocess.run(shlex.split('python setup.py install'), cwd=os.path.join(root, 'submodules', 'simple-knn'))