Update load_models.py

load_models.py

@@ -2,8 +2,8 @@ import sys
 import os
 import numpy as np
 import torch
-from ultralytics import YOLO
 from PIL import Image
+from huggingface_hub import hf_hub_download
 
 # StyleGAN2 dependencies
 sys.path.append("/Users/sophiemaw/stylegan2-ada")
@@ -20,6 +20,12 @@ def load_stylegan_model(
 ) -> None:
     global stylegan_generator
 
+    print("⬇️ Downloading StyleGAN2 model from Hugging Face Hub...")
+    model_path = hf_hub_download(
+        repo_id="ZeqiangLai/StyleGAN2",
+        filename="stylegan2-ffhq-config-f.pkl"
+    )
+
     print(f"🔄 Loading StyleGAN2 model from: {model_path}")
     with open(model_path, 'rb') as f:
         G = legacy.load_network_pkl(f)['G_ema'].to(device)