IMAGE_CAPTION_MODEL = "Salesforce/blip-image-captioning-large"

LLM_MODEL = "microsoft/Phi-3-mini-4k-instruct"
LLM_MAX_LENGTH = 50
LLM_MAX_NEW_TOKENS = 50

MUSICGEN_MODEL = "facebook/musicgen-small"
MUSICGEN_MAX_NEW_TOKENS = 256 # 5 seconds of audio

AUDIO_DIR = "Case-Study-1/data/"