torch
torchvision
matplotlib
Pillow
numpy==1.26.4
ipykernel
ipywidgets
ipympl

# Pin these to compatible versions
huggingface_hub<1.0.0
transformers>=4.40.0,<5.0.0