drrobot9's picture
Upload folder using huggingface_hub
2b77fb1 verified
Raw History Blame Contribute Delete
869 Bytes
import os
MODEL_REPO_ID = "drrobot9/sign-to-text-shubert-weights"
MODELS_DIR = "/app/models"
POSE_LANDMARKER_PATH = os.path.join(MODELS_DIR, "pose_landmarker_lite.task")
HAND_LANDMARKER_PATH = os.path.join(MODELS_DIR, "hand_landmarker.task")
FACE_LANDMARKER_PATH = os.path.join(MODELS_DIR, "face_landmarker_v2_with_blendshapes.task")
SHUBERT_CHECKPOINT_PATH = os.path.join(MODELS_DIR, "checkpoint_836_400000.pt")
DINO_HAND_CHECKPOINT_PATH = os.path.join(MODELS_DIR, "dinov2hand.pth")
DINO_FACE_CHECKPOINT_PATH = os.path.join(MODELS_DIR, "dinov2face.pth")
BYT5_CHECKPOINT_DIR = os.path.join(MODELS_DIR, "checkpoint-11625")
BYT5_BASE_DIR = os.path.join(MODELS_DIR, "byt5_base")
DEVICE = "cuda"
HANDS_LOW_THRESHOLD = 1.2
STILL_SPEED_THRESHOLD = 0.3
START_HOLD_SECONDS = 0.2
REST_HOLD_SECONDS = 0.8
MIN_SEGMENT_SECONDS = 0.8
MAX_SEGMENT_SECONDS = 15.0
TARGET_FPS = 15