Update app.py

app.py

@@ -28,7 +28,6 @@ except ImportError:
     AttnProcessor2_0 = None
 from transformers import CLIPTextModel, CLIPTokenizer
 from enum import Enum
-from torch.hub import download_url_to_file
 import torch.nn as nn
 import torch.nn.functional as F
 from huggingface_hub import PyTorchModelHubMixin
@@ -134,7 +133,19 @@ unet.forward = hooked_unet_forward
 model_path = './iclight_sd15_fbc.safetensors'
 if not os.path.exists(model_path):
     print("Downloading IC-Light model...")
-    download_url_to_file(url='https://huggingface.co/lllyasviel/ic-light/resolve/main/iclight_sd15_fbc.safetensors', dst=model_path)
+    try:
+        from huggingface_hub import hf_hub_download
+        model_path = hf_hub_download(
+            repo_id="lllyasviel/ic-light",
+            filename="iclight_sd15_fbc.safetensors",
+            local_dir="./",
+            local_dir_use_symlinks=False
+        )
+    except Exception as e:
+        print(f"Failed to download with hf_hub_download: {e}")
+        # Fallback to torch.hub
+        from torch.hub import download_url_to_file
+        download_url_to_file(url='https://huggingface.co/lllyasviel/ic-light/resolve/main/iclight_sd15_fbc.safetensors', dst=model_path)
 
 sd_offset = sf.load_file(model_path)
 sd_origin = unet.state_dict()