fix: set token

app.py

@@ -41,6 +41,7 @@ logger = logging.getLogger(__name__)
 env = os.environ.get('COIN_ENV', 'local')
 if env == 'hf':
     config_fp = 'hf_config.yaml'
+    hf_token = os.environ.get('HF_TOKEN')
 else:
     config_fp = 'local_config.yaml'
 logger.info(f'Use config file: {config_fp}')
@@ -57,15 +58,22 @@ RESIZED_TO_BEFORE_DETECT = total_config['detector'].get('resized_to', 300)
 def prepare_chromadb():
     if env == 'local':
         return
+
     from huggingface_hub import snapshot_download
-    snapshot_download(repo_type='model', repo_id='breezedeus/usa-coins-chromadb', local_dir='./')
+
+    snapshot_download(
+        repo_type='model',
+        repo_id='breezedeus/usa-coins-chromadb',
+        local_dir='./',
+        token=hf_token,
+    )
 
 
 def _load_dataset(data_path):
     logger.info('Load dataset from %s', data_path)
 
     if env == 'hf':
-        dataset = load_dataset(data_path, split='train')
+        dataset = load_dataset(data_path, split='train', token=hf_token)
     else:
         dataset = load_dataset("imagefolder", data_dir=data_path, split='train')
     return dataset
@@ -107,8 +115,7 @@ def load_chroma_db(db_dir, collection_name, model_name, device='cpu'):
 
     embedding_function = ChromaEmbeddingFunction(model_name, device)
     collection = client.get_collection(
-        name=collection_name,
-        embedding_function=embedding_function,
+        name=collection_name, embedding_function=embedding_function,
     )
     return collection
 
@@ -211,13 +218,21 @@ def main():
             with gr.Column(variant='compact', scale=1):
                 gr.Markdown('### Results from Coin-CLIP')
                 cc_results = gr.Gallery(
-                    label='Coin-CLIP Results', columns=3, height=2200, show_share_button=True, visible=False
+                    label='Coin-CLIP Results',
+                    columns=3,
+                    height=2200,
+                    show_share_button=True,
+                    visible=False,
                 )
 
             with gr.Column(variant='compact', scale=1):
                 gr.Markdown('### Results from CLIP')
                 coin_results = gr.Gallery(
-                    label='CLIP Results', columns=3, height=2200, show_share_button=True, visible=False
+                    label='CLIP Results',
+                    columns=3,
+                    height=2200,
+                    show_share_button=True,
+                    visible=False,
                 )
 
             sub_btn.click(