Update app.py

app.py

@@ -41,6 +41,54 @@ import torch
 from PIL import Image
 
 
+# Load
+
+# Model paths
+model_path = './models/iclight_sd15_fc.safetensors'
+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()
+
+
+
 
 try:
     import xformers
@@ -134,51 +182,6 @@ def hooked_unet_forward(sample, timestep, encoder_hidden_states, **kwargs):
 
 unet.forward = hooked_unet_forward
 
-# Load
-
-# Model paths
-model_path = './models/iclight_sd15_fc.safetensors'
-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()