# ML / DL Frameworks
torch==2.4.0
torchvision==0.19
transformers

# 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.0
matplotlib

# MLOps & Infrastructure
wandb==0.16.6

# Utilities
PyYAML==6.0.2
tqdm==4.67.1
albumentations
ftfy
open-clip-torch

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