# ML / DL Frameworks
torch>=2.5.0
torchvision>=0.20.0
transformers>=4.57,<5  # PGC official weights need the transformers 4.x DINOv2 key names

# ML/DL Helpers
einops==0.8.0

# Data Science
# Using headless opencv to prevent libGL errors in Docker/Spaces
opencv-python-headless==4.10.0.84
pandas==2.2.3
scikit-image==0.22.0
scikit-learn==1.5.2
numpy<2
matplotlib

# MLOps & Infrastructure
wandb==0.16.6

# Utilities
PyYAML==6.0.2
tqdm==4.67.1
albumentations>=1.4.24,<2
kornia>=0.8,<0.9
scipy
huggingface_hub
ftfy
open-clip-torch

# 1. Use latest Gradio
gradio==5.49.1
# 2. Unpin FastAPI and Pydantic to allow latest versions
# fastapi
# pydantic