requirements.txt

PyQt6>=6.6.0
PyQt6-WebEngine>=6.6.0
Pillow>=10.0.0
numpy>=1.24.0
huggingface-hub>=0.19.0
pandas>=2.0.0
tqdm>=4.66.0

# CLIP Interrogator
clip-interrogator>=0.6.0

# Note: PyTorch and ONNX Runtime are handled by setup scripts
# The setup script will automatically install:
#   - PyTorch with CUDA (if NVIDIA GPU detected), PyTorch with ROCm (if AMD GPU detected on linux) or PyTorch (for CPU-only)
#   - onnxruntime-gpu (if GPU detected on x86_64) or onnxruntime (CPU-only)
#
# ARM64 Note: onnxruntime-gpu wheels are not available on PyPI for ARM64.
# WD Tagger will run on CPU. For GPU support, build onnxruntime from source.
# See: https://onnxruntime.ai/docs/build/eps.html#cuda
#
# Manual installation (x86_64):
#   GPU: pip install torch torchvision --index-url https://download.pytorch.org/whl/cu126
#        pip install onnxruntime-gpu>=1.16.0
#   CPU: pip install torch torchvision
#        pip install onnxruntime>=1.16.0