Update src/load_models.py

src/load_models.py

@@ -1,19 +1,14 @@
 import sys
 import os
-import subprocess
 import torch
 import numpy as np
 from PIL import Image
 from huggingface_hub import hf_hub_download
 
-# Clone StyleGAN2 repo if not already present
-if not os.path.exists("stylegan2-ada"):
-    subprocess.run(["git", "clone", "https://github.com/NVlabs/stylegan2-ada-pytorch.git", "stylegan2-ada"])
-sys.path.append("stylegan2-ada")
+# Point to your local stylegan2 utilities
+#sys.path.append("src/stylegan2")  # Adjust if the path differs
 
-# Now import from the cloned repo
-import dnnlib
-import legacy
+from stylegan2 import legacy
 
 # Globals
 stylegan_generator = None
@@ -37,13 +32,11 @@ def load_stylegan_model(device="cuda" if torch.cuda.is_available() else "cpu"):
     stylegan_generator = G
     print("✅ StyleGAN2 model loaded and ready.")
 
-
 def warm_up_models():
     """
     Loads and warms up all global models used in the pipeline.
     """
     global stylegan_generator
 
-
     if stylegan_generator is None:
         load_stylegan_model()