breezedeus commited on
Commit
9d54244
1 Parent(s): 5315596

fix: name conflict

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 load_dataset(data_path):
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 = load_dataset(**total_config['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()