Spaces:
Running
Running
Commit ·
6fef2e8
1
Parent(s): b137359
hf token added
Browse files- api/utils.py +8 -1
api/utils.py
CHANGED
|
@@ -26,10 +26,17 @@ from .config import (
|
|
| 26 |
ALLOWED_EXTENSIONS,
|
| 27 |
MIN_IMAGES,
|
| 28 |
MAX_IMAGES,
|
| 29 |
-
UPLOAD_DIR,
|
| 30 |
logger,
|
| 31 |
)
|
|
|
|
| 32 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
# ----------------------------------------------------------------
|
| 34 |
# BUCKET SETUP
|
| 35 |
# All data is stored under:
|
|
|
|
| 26 |
ALLOWED_EXTENSIONS,
|
| 27 |
MIN_IMAGES,
|
| 28 |
MAX_IMAGES,
|
| 29 |
+
UPLOAD_DIR,
|
| 30 |
logger,
|
| 31 |
)
|
| 32 |
+
from huggingface_hub import login
|
| 33 |
|
| 34 |
+
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 35 |
+
|
| 36 |
+
if HF_TOKEN:
|
| 37 |
+
login(token=HF_TOKEN)
|
| 38 |
+
else:
|
| 39 |
+
print("❌ HF_TOKEN missing")
|
| 40 |
# ----------------------------------------------------------------
|
| 41 |
# BUCKET SETUP
|
| 42 |
# All data is stored under:
|