# Core ML libraries
torch>=2.0.0,<3.0.0
torchvision>=0.15.0,<1.0.0
ultralytics>=8.0.0,<9.0.0
transformers>=4.30.0,<5.0.0
Pillow>=9.0.0,<11.0.0

# Computer Vision - ONLY headless version, no regular opencv
opencv-python-headless==4.8.1.78

# Vector search
faiss-cpu>=1.7.0,<2.0.0
# For GPU support, use: faiss-gpu>=1.7.0

# Utilities
numpy>=1.24.0,<2.0.0
scikit-image>=0.20.0,<1.0.0
tqdm>=4.65.0,<5.0.0

# FastAPI and server
fastapi>=0.100.0,<1.0.0
uvicorn>=0.20.0,<1.0.0
python-multipart>=0.0.5

# HTTP client for testing
requests>=2.28.0,<3.0.0

# Optional: For better performance
# tensorrt  # Only if you have NVIDIA GPU
# onnxruntime-gpu  # Only if you have NVIDIA GPU