Spaces:
Running
Running
File size: 722 Bytes
4ab551f af5dd9a 4ab551f 034aa20 4ab551f 8f639d1 4ab551f 0a44e9a 4ab551f 0a44e9a 5ffc380 4ab551f 5ffc380 4ab551f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
from pathlib import Path
ROOT = Path(__file__).parent.parent
RESOURCES_DIR = Path("/tmp/resources")
WATER_MARK_TEMPLATE_IMAGE_PATH = RESOURCES_DIR / "watermark_template.png"
WATER_MARK_DETECT_YOLO_WEIGHTS = RESOURCES_DIR / "best.pt"
WATER_MARK_DETECT_YOLO_WEIGHTS_HASH_JSON = RESOURCES_DIR / "model_version.json"
OUTPUT_DIR = Path("/tmp/output")
OUTPUT_DIR.mkdir(exist_ok=True, parents=True)
DEFAULT_WATERMARK_REMOVE_MODEL = "lama"
WORKING_DIR = Path("/tmp/working_dir")
WORKING_DIR.mkdir(exist_ok=True, parents=True)
LOGS_PATH = Path("/tmp/logs")
LOGS_PATH.mkdir(exist_ok=True, parents=True)
DATA_PATH = Path("/tmp/data")
DATA_PATH.mkdir(exist_ok=True, parents=True)
SQLITE_PATH = DATA_PATH / "db.sqlite3"
|