Spaces:
Sleeping
Sleeping
| import os | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| DEFAULT_PINECONE_KEY = os.getenv("DEFAULT_PINECONE_KEY", "") | |
| DEFAULT_CLOUDINARY_URL = os.getenv("DEFAULT_CLOUDINARY_URL", "") | |
| IDX_FACES = os.getenv("IDX_FACES", "enterprise-faces") | |
| IDX_OBJECTS = os.getenv("IDX_OBJECTS", "enterprise-objects") | |
| MAX_CONCURRENT_INFERENCES = int(os.getenv("MAX_CONCURRENT_INFERENCES", "1")) | |
| MAX_FILES_PER_UPLOAD = int(os.getenv("MAX_FILES_PER_UPLOAD", "50")) | |
| INFERENCE_CACHE_SIZE = int(os.getenv("INFERENCE_CACHE_SIZE", "128")) | |
| SUPABASE_URL = os.getenv("SUPABASE_URL", "") | |
| SUPABASE_SERVICE_KEY = os.getenv("SUPABASE_SERVICE_KEY", "") | |
| MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1024")) | |
| MAX_CROPS = int(os.getenv("MAX_CROPS", "10")) | |
| YOLO_PERSON_CLASS_ID = 0 | |
| YOLO_MIN_CROP_PX = int(os.getenv("YOLO_MIN_CROP_PX", "50")) | |
| YOLO_CONF_THRESHOLD = float(os.getenv("YOLO_CONF_THRESHOLD", "0.25")) | |
| DET_SIZE_PRIMARY = (640, 640) | |
| DET_SCALES = [(1280, 1280), (960, 960), (640, 640)] | |
| IOU_DEDUP_THRESHOLD = float(os.getenv("IOU_DEDUP_THRESHOLD", "0.4")) | |
| MIN_FACE_SIZE = int(os.getenv("MIN_FACE_SIZE", "30")) | |
| MAX_FACES_PER_IMAGE = int(os.getenv("MAX_FACES_PER_IMAGE", "20")) | |
| FACE_QUALITY_GATE = float(os.getenv("FACE_QUALITY_GATE", "0.5")) | |
| FACE_DIM = 512 | |
| ADAFACE_DIM = 512 | |
| FUSED_FACE_DIM = 1024 | |
| FACE_CROP_THUMB_SIZE = int(os.getenv("FACE_CROP_THUMB_SIZE", "112")) | |
| FACE_CROP_QUALITY = int(os.getenv("FACE_CROP_QUALITY", "85")) | |
| FACE_CROP_PADDING = float(os.getenv("FACE_CROP_PADDING", "0.2")) | |
| ADAFACE_CROP_PADDING = float(os.getenv("ADAFACE_CROP_PADDING", "0.1")) | |
| ENABLE_ADAFACE = int(os.getenv("ENABLE_ADAFACE", "1")) | |
| HF_TOKEN = os.getenv("HF_TOKEN", "") |