# This file lists the required packages for the clip-face-attribute-classifier project.
# Install them using: pip install -r requirements.txt

# --- Hugging Face Libraries ---
# Core library for models, Trainer, TrainingArguments, and processors
transformers==4.38.2
# Used for data handling and creating Dataset objects
datasets==2.18.0
# For efficient training and hardware acceleration with the Trainer
accelerate==0.27.2
# For interacting with the Hugging Face Hub (login, upload, etc.)
huggingface_hub==0.21.4


# --- Core Deep Learning Framework ---
# The fundamental deep learning library
torch==2.2.1
# Companion library for computer vision tasks in PyTorch
torchvision==0.17.1


# --- Data Handling and Metrics ---
# For reading and manipulating the .csv label files
pandas==2.2.1
# For calculating evaluation metrics like accuracy, precision, recall, and F1-score
scikit-learn==1.4.1.post1


# --- Utilities ---
# For opening and handling image files
Pillow==10.2.0
# For creating progress bars during evaluation
tqdm==4.66.2
# For loading the safer .safetensors model format
safetensors==0.4.2