codewithRiz commited on
Commit
6fef2e8
·
1 Parent(s): b137359

hf token added

Browse files
Files changed (1) hide show
  1. 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, # e.g. "codewithRiz/test_bucket"
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: