Spaces:
Running
Running
Commit ·
dcba3cd
1
Parent(s): 4f1c854
bucket secret fix
Browse files- 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 |
-
|
| 25 |
-
|
| 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"}
|