Spaces:
Running
Running
breezedeus
commited on
Commit
•
9d54244
1
Parent(s):
5315596
fix: name conflict
Browse files
app.py
CHANGED
@@ -61,7 +61,7 @@ def prepare_chromadb():
|
|
61 |
snapshot_download(repo_type='model', repo_id='breezedeus/usa-coins-chromadb', local_dir='./')
|
62 |
|
63 |
|
64 |
-
def
|
65 |
logger.info('Load dataset from %s', data_path)
|
66 |
|
67 |
if env == 'hf':
|
@@ -123,7 +123,7 @@ def retrieve(query_image: Image.Image, collection, top_k=20) -> List[Image.Image
|
|
123 |
return [ds_dict[id]['image'] for id in retrieved['ids'][0]]
|
124 |
|
125 |
|
126 |
-
dataset =
|
127 |
ds_dict = {_d['id']: _d for _d in dataset}
|
128 |
|
129 |
prepare_chromadb()
|
|
|
61 |
snapshot_download(repo_type='model', repo_id='breezedeus/usa-coins-chromadb', local_dir='./')
|
62 |
|
63 |
|
64 |
+
def _load_dataset(data_path):
|
65 |
logger.info('Load dataset from %s', data_path)
|
66 |
|
67 |
if env == 'hf':
|
|
|
123 |
return [ds_dict[id]['image'] for id in retrieved['ids'][0]]
|
124 |
|
125 |
|
126 |
+
dataset = _load_dataset(**total_config['dataset'])
|
127 |
ds_dict = {_d['id']: _d for _d in dataset}
|
128 |
|
129 |
prepare_chromadb()
|