Update gradio_demo.py

gradio_demo.py

@@ -141,12 +141,53 @@ unet.forward = hooked_unet_forward
 
 # Load
 
+# Model paths
 model_path = './models/iclight_sd15_fc.safetensors'
-#model_path = './models/iclight_sd15_fbc.safetensors'
-
-
-# if not os.path.exists(model_path):
-#     download_url_to_file(url='https://huggingface.co/lllyasviel/ic-light/resolve/main/iclight_sd15_fc.safetensors', dst=model_path)
+model_path2 = './checkpoints/depth_anything_v2_vits.pth'
+model_path3 = './checkpoints/sam2_hiera_large.pt'
+model_path4 = './checkpoints/config.json'
+model_path5 = './checkpoints/preprocessor_config.json'
+model_path6 = './configs/sam2_hiera_l.yaml'
+model_path7 = './mvadapter_i2mv_sdxl.safetensors'
+
+# Base URL for the repository
+BASE_URL = 'https://huggingface.co/Ashoka74/Placement/resolve/main/'
+
+# Model URLs
+model_urls = {
+    model_path: 'iclight_sd15_fc.safetensors',
+    model_path2: 'depth_anything_v2_vits.pth',
+    model_path3: 'sam2_hiera_large.pt',
+    model_path4: 'config.json',
+    model_path5: 'preprocessor_config.json',
+    model_path6: 'sam2_hiera_l.yaml',
+    model_path7: 'mvadapter_i2mv_sdxl.safetensors'
+}
+
+# Ensure directories exist
+def ensure_directories():
+    for path in model_urls.keys():
+        os.makedirs(os.path.dirname(path), exist_ok=True)
+
+# Download models
+def download_models():
+    for local_path, filename in model_urls.items():
+        if not os.path.exists(local_path):
+            try:
+                url = f"{BASE_URL}{filename}"
+                print(f"Downloading {filename}")
+                download_url_to_file(url, local_path)
+                print(f"Successfully downloaded {filename}")
+            except Exception as e:
+                print(f"Error downloading {filename}: {e}")
+
+ensure_directories()
+
+download_models()
+
+
+if not os.path.exists(model_path):
+    download_url_to_file(url='https://huggingface.co/lllyasviel/ic-light/resolve/main/iclight_sd15_fc.safetensors', dst=model_path)
 
 sd_offset = sf.load_file(model_path)
 sd_origin = unet.state_dict()