Update run_gradio.py

run_gradio.py

@@ -27,6 +27,52 @@ from src.models.refunet_2d_condition import RefUNet2DConditionModel
 from src.point_network import PointNet
 from src.annotator.lineart import BatchLineartDetector
 val_configs = OmegaConf.load('./configs/inference.yaml')
+# download the checkpoints
+from huggingface_hub import snapshot_download, hf_hub_download
+
+os.makedirs("checkpoints", exist_ok=True)
+
+# List of subdirectories to create inside "checkpoints"
+subfolders = [
+    "StableDiffusion",
+    "models",
+    "MangaNinja"
+]
+# Create each subdirectory
+for subfolder in subfolders:
+    os.makedirs(os.path.join("checkpoints", subfolder), exist_ok=True)
+
+# List of subdirectories to create inside "models"
+models_subfolders = [
+    "clip-vit-large-patch14",
+    "control_v11p_sd15_lineart",
+    "Annotators"
+]
+# Create each subdirectory
+for subfolder in models_subfolders:
+    os.makedirs(os.path.join("checkpoints/models", subfolder), exist_ok=True)
+
+snapshot_download(
+    repo_id = "stable-diffusion-v1-5/stable-diffusion-v1-5",
+    local_dir = "./checkpoints/StableDiffusion"  
+)
+snapshot_download(
+    repo_id = "openai/clip-vit-large-patch14",
+    local_dir = "./checkpoints/models/clip-vit-large-patch14"  
+)
+snapshot_download(
+    repo_id = "lllyasviel/control_v11p_sd15_lineart",
+    local_dir = "./checkpoints/models/control_v11p_sd15_lineart"  
+)
+hf_hub_download(
+    repo_id = "lllyasviel/Annotators",
+    filename = "sk_model.pth",
+    local_dir = "./checkpoints/models/Annotators" 
+)
+snapshot_download(
+    repo_id = "Johanan0528/MangaNinjia",
+    local_dir = "./checkpoints/MangaNinja"  
+)
 # === load the checkpoint ===
 pretrained_model_name_or_path = val_configs.model_path.pretrained_model_name_or_path
 refnet_clip_vision_encoder_path = val_configs.model_path.clip_vision_encoder_path
@@ -378,4 +424,4 @@ with gr.Blocks() as demo:
                            outputs=[baseline_gallery]
     )
 
-demo.launch(server_name="0.0.0.0")
+demo.launch(show_api=False, show_error=True)