bug-fix:load spiga ckpt from local

checkpoints/spiga_300wpublic.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:98f014611ac25d549e89083992d9e6ade15da133c634a3883473abf2953cee2d
+size 254397265

inference_utils.py

@@ -10,17 +10,6 @@ torch.cuda.manual_seed_all(seed)
 torch.backends.cudnn.deterministic = True
 torch.backends.cudnn.benchmark = False
 
-# SPIGA ckpt downloading always fails, so we download it manually and put it in the right place.
-import spiga
-from gdown import download
-
-pkg_path = spiga.__file__.replace("/__init__.py", "")
-spiga_file_id = "1YrbScfMzrAAWMJQYgxdLZ9l57nmTdpQC"
-ckpt_path = os.path.join(pkg_path, "spiga/models/weights/spiga_300wpublic.pt")
-if not os.path.exists(ckpt_path):
-    os.makedirs(os.path.dirname(ckpt_path), exist_ok=True)
-    download(id=spiga_file_id, output=ckpt_path)
-
 from PIL import Image
 from gdown import download_folder
 from facelib import FaceDetector

spiga_draw.py

@@ -7,7 +7,16 @@ from facelib import FaceDetector
 from spiga.inference.config import ModelConfig
 from spiga.inference.framework import SPIGAFramework
 
-processor = SPIGAFramework(ModelConfig("300wpublic"))
+# SPIGA ckpt downloading always fails, so we load it from the local path instead.
+spiga_ckpt = os.path.join(os.path.dirname(__file__), "checkpoints/spiga_300wpublic.pt")
+if not os.path.exists(spiga_ckpt):
+    from gdown import download
+    spiga_file_id = "1YrbScfMzrAAWMJQYgxdLZ9l57nmTdpQC"
+    download(id=spiga_file_id, output=spiga_ckpt)
+spiga_config = ModelConfig("300wpublic")
+spiga_config.load_model_url = False
+spiga_config.model_weights_path = os.path.dirname(spiga_ckpt)
+processor = SPIGAFramework(spiga_config)
 
 def center_crop(image, size):
     width, height = image.size