fix: add HF token Authorization header for model downloads from huggingface.co

DeepFakeAI/utilities.py

@@ -7,6 +7,7 @@ import ssl
 import subprocess
 import tempfile
 import urllib
+from urllib.request import Request, urlopen, urlretrieve
 from pathlib import Path
 from typing import List, Optional
 
@@ -170,10 +171,17 @@ def conditional_download(download_directory_path : str, urls : List[str]) -> Non
         if os.path.exists(download_file_path):
             return
         try:
-            request = urllib.request.urlopen(url)  # type: ignore[attr-defined]
-            total = int(request.headers.get('Content-Length', 0) or 0)
+            headers = {}
+            # If downloading from Hugging Face, include token when available
+            if 'huggingface.co' in url:
+                token = os.environ.get('TOKEN') or os.environ.get('HF_TOKEN')
+                if token:
+                    headers['Authorization'] = f'Bearer {token}'
+            req: Request = Request(url, headers=headers)
+            resp = urlopen(req)  # type: ignore[assignment]
+            total = int(resp.headers.get('Content-Length', 0) or 0)
             with tqdm(total = total, desc = wording.get('downloading'), unit = 'B', unit_scale = True, unit_divisor = 1024) as progress:
-                urllib.request.urlretrieve(url, download_file_path, reporthook = lambda count, block_size, total_size: progress.update(block_size))  # type: ignore[attr-defined]
+                urlretrieve(url, download_file_path, reporthook = lambda count, block_size, total_size: progress.update(block_size))  # type: ignore[attr-defined]
             return
         except Exception as e:  # try next mirror
             last_error = e