Try fix 4
Browse files- backend/runner/config.py +25 -5
backend/runner/config.py
CHANGED
|
@@ -103,7 +103,11 @@ def load_json_from_hf(repo_id: str, filename: str) -> Optional[Dict[str, Any]]:
|
|
| 103 |
|
| 104 |
try:
|
| 105 |
print(f"🔍 Downloading {filename} from {repo_id}...")
|
| 106 |
-
file_path = hf_hub_download(
|
|
|
|
|
|
|
|
|
|
|
|
|
| 107 |
|
| 108 |
with open(file_path, 'r', encoding='utf-8') as f:
|
| 109 |
data = json.load(f)
|
|
@@ -160,10 +164,26 @@ def load_embeddings_datasets() -> Optional[Dict[str, Any]]:
|
|
| 160 |
print(f" Loading embeddings files from {ARTEFACT_EMBEDDINGS_DATASET}...")
|
| 161 |
|
| 162 |
# Download the files to local paths
|
| 163 |
-
clip_embeddings_path = hf_hub_download(
|
| 164 |
-
|
| 165 |
-
|
| 166 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 167 |
|
| 168 |
print(f"✅ Successfully downloaded embeddings files:")
|
| 169 |
print(f" CLIP embeddings: {clip_embeddings_path}")
|
|
|
|
| 103 |
|
| 104 |
try:
|
| 105 |
print(f"🔍 Downloading {filename} from {repo_id}...")
|
| 106 |
+
file_path = hf_hub_download(
|
| 107 |
+
repo_id=repo_id,
|
| 108 |
+
filename=filename,
|
| 109 |
+
repo_type="dataset"
|
| 110 |
+
)
|
| 111 |
|
| 112 |
with open(file_path, 'r', encoding='utf-8') as f:
|
| 113 |
data = json.load(f)
|
|
|
|
| 164 |
print(f" Loading embeddings files from {ARTEFACT_EMBEDDINGS_DATASET}...")
|
| 165 |
|
| 166 |
# Download the files to local paths
|
| 167 |
+
clip_embeddings_path = hf_hub_download(
|
| 168 |
+
repo_id=ARTEFACT_EMBEDDINGS_DATASET,
|
| 169 |
+
filename='clip_embeddings.safetensors',
|
| 170 |
+
repo_type="dataset"
|
| 171 |
+
)
|
| 172 |
+
paintingclip_embeddings_path = hf_hub_download(
|
| 173 |
+
repo_id=ARTEFACT_EMBEDDINGS_DATASET,
|
| 174 |
+
filename='paintingclip_embeddings.safetensors',
|
| 175 |
+
repo_type="dataset"
|
| 176 |
+
)
|
| 177 |
+
clip_sentence_ids_path = hf_hub_download(
|
| 178 |
+
repo_id=ARTEFACT_EMBEDDINGS_DATASET,
|
| 179 |
+
filename='clip_embeddings_sentence_ids.json',
|
| 180 |
+
repo_type="dataset"
|
| 181 |
+
)
|
| 182 |
+
paintingclip_sentence_ids_path = hf_hub_download(
|
| 183 |
+
repo_id=ARTEFACT_EMBEDDINGS_DATASET,
|
| 184 |
+
filename='paintingclip_embeddings_sentence_ids.json',
|
| 185 |
+
repo_type="dataset"
|
| 186 |
+
)
|
| 187 |
|
| 188 |
print(f"✅ Successfully downloaded embeddings files:")
|
| 189 |
print(f" CLIP embeddings: {clip_embeddings_path}")
|