Update predict_blend.py

predict_blend.py

@@ -5,13 +5,12 @@ from huggingface_hub import snapshot_download
 from pathlib import Path
 from itertools import product
 
-# =========================
-# Config
-# =========================
 MODEL_REPO = "Antonio0616/FormingStar"
-MODEL_DIR = snapshot_download(repo_id=MODEL_REPO)
-ART_DIR = str(Path(MODEL_DIR).resolve())
 
+# ✅ 반드시 dataset 으로 명시해야 함
+
+MODEL_DIR = snapshot_download(repo_id=MODEL_REPO, repo_type="dataset")
+ART_DIR = str(Path(MODEL_DIR).resolve())
 CAT_COL = "material"
 NUM_COLS = ["thickness","diameter","degree","upper_radius","lower_radius","LB","RB"]