codewithRiz commited on
Commit
dcba3cd
·
1 Parent(s): 4f1c854

bucket secret fix

Browse files
Files changed (1) hide show
  1. api/config.py +4 -5
api/config.py CHANGED
@@ -20,13 +20,12 @@ logger = logging.getLogger("WildlifeLogger")
20
 
21
  ENV = os.getenv("ENV", "DEV").upper()
22
  logger.info(f"Running in {ENV}")
 
23
 
24
- # ---------------- STORAGE ----------------
25
- # Use Hugging Face bucket as UPLOAD_DIR
26
- UPLOAD_DIR = "codewithRiz/test_bucket"
27
- STORAGE_BACKEND = "huggingface" # indicate we are using HF bucket
28
 
29
- # ---------------- UPLOAD RULES ----------------
30
  MIN_IMAGES = 1
31
  MAX_IMAGES = 1000
32
  ALLOWED_EXTENSIONS = {"jpg", "jpeg", "png", "webp"}
 
20
 
21
  ENV = os.getenv("ENV", "DEV").upper()
22
  logger.info(f"Running in {ENV}")
23
+ UPLOAD_DIR = os.getenv("HF_BUCKET_ID")
24
 
25
+ if not UPLOAD_DIR:
26
+ raise ValueError("HF_BUCKET_ID is not set in environment variables")
 
 
27
 
28
+ # ---------------- UPLOAD RULES --------w--------
29
  MIN_IMAGES = 1
30
  MAX_IMAGES = 1000
31
  ALLOWED_EXTENSIONS = {"jpg", "jpeg", "png", "webp"}