Upload app.py with huggingface_hub

app.py

@@ -10,11 +10,25 @@
 # its affiliates is strictly prohibited.
 #
 # For permission requests, please contact robot0321@snu.ac.kr, esw0116@snu.ac.kr, namhj28@gmail.com, jarin.lee@gmail.com.
+import os
 import time
+import pathlib
 import gradio as gr
+from huggingface_hub import snapshot_download
 from luciddreamer import LucidDreamer
 
 
+root = pathlib.Path(__file__).parent
+example_root = os.path.joint(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'))
+
+
 css = """
 #run-button {
   background: coral;
@@ -24,7 +38,6 @@ css = """
 
 ld = LucidDreamer()
 
-
 with gr.Blocks(css=css) as demo:
 
     gr.HTML(
@@ -265,4 +278,4 @@ with gr.Blocks(css=css) as demo:
 
 
 if __name__ == '__main__':
-    demo.queue(max_size=20).launch(server_port=16011)
+    demo.queue(max_size=20).launch(server_port=16011)