Fix token error for public model download
Browse files- download_models.py +1 -1
download_models.py
CHANGED
|
@@ -27,7 +27,7 @@ def download_models():
|
|
| 27 |
logger.info(f"Downloading audio model: {HF_AUDIO_MODEL}...")
|
| 28 |
try:
|
| 29 |
# Load pipeline to force download, providing token for gated models
|
| 30 |
-
hf_token = os.environ.get("HUGGING_FACE_TOKEN")
|
| 31 |
_ = pipeline("audio-classification", model=HF_AUDIO_MODEL, token=hf_token)
|
| 32 |
logger.info("Successfully cached audio model.")
|
| 33 |
except Exception as e:
|
|
|
|
| 27 |
logger.info(f"Downloading audio model: {HF_AUDIO_MODEL}...")
|
| 28 |
try:
|
| 29 |
# Load pipeline to force download, providing token for gated models
|
| 30 |
+
hf_token = os.environ.get("HUGGING_FACE_TOKEN", "").strip() or None
|
| 31 |
_ = pipeline("audio-classification", model=HF_AUDIO_MODEL, token=hf_token)
|
| 32 |
logger.info("Successfully cached audio model.")
|
| 33 |
except Exception as e:
|