samwaugh commited on
Commit
a02b702
·
1 Parent(s): 61f8035

Try fix 4

Browse files
Files changed (1) hide show
  1. 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(repo_id=repo_id, filename=filename)
 
 
 
 
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(repo_id=ARTEFACT_EMBEDDINGS_DATASET, filename='clip_embeddings.safetensors')
164
- paintingclip_embeddings_path = hf_hub_download(repo_id=ARTEFACT_EMBEDDINGS_DATASET, filename='paintingclip_embeddings.safetensors')
165
- clip_sentence_ids_path = hf_hub_download(repo_id=ARTEFACT_EMBEDDINGS_DATASET, filename='clip_embeddings_sentence_ids.json')
166
- paintingclip_sentence_ids_path = hf_hub_download(repo_id=ARTEFACT_EMBEDDINGS_DATASET, filename='paintingclip_embeddings_sentence_ids.json')
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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}")