Rename

app.py

@@ -19,7 +19,7 @@ TITLE = 'KichangKim/DeepDanbooru'
 DESCRIPTION = 'This is an unofficial demo for https://github.com/KichangKim/DeepDanbooru.'
 ARTICLE = '<center><img src="https://visitor-badge.glitch.me/badge?page_id=hysts.deepdanbooru" alt="visitor badge"/></center>'
 
-TOKEN = os.environ['TOKEN']
+HF_TOKEN = os.environ['HF_TOKEN']
 MODEL_REPO = 'hysts/DeepDanbooru'
 MODEL_FILENAME = 'model-resnet_custom_v3.h5'
 LABEL_FILENAME = 'tags.txt'
@@ -47,7 +47,7 @@ def load_sample_image_paths() -> list[pathlib.Path]:
         path = huggingface_hub.hf_hub_download(dataset_repo,
                                                'images.tar.gz',
                                                repo_type='dataset',
-                                               use_auth_token=TOKEN)
+                                               use_auth_token=HF_TOKEN)
         with tarfile.open(path) as f:
             f.extractall()
     return sorted(image_dir.glob('*'))
@@ -56,7 +56,7 @@ def load_sample_image_paths() -> list[pathlib.Path]:
 def load_model() -> tf.keras.Model:
     path = huggingface_hub.hf_hub_download(MODEL_REPO,
                                            MODEL_FILENAME,
-                                           use_auth_token=TOKEN)
+                                           use_auth_token=HF_TOKEN)
     model = tf.keras.models.load_model(path)
     return model
 
@@ -64,7 +64,7 @@ def load_model() -> tf.keras.Model:
 def load_labels() -> list[str]:
     path = huggingface_hub.hf_hub_download(MODEL_REPO,
                                            LABEL_FILENAME,
-                                           use_auth_token=TOKEN)
+                                           use_auth_token=HF_TOKEN)
     with open(path) as f:
         labels = [line.strip() for line in f.readlines()]
     return labels