python -m venv venv
source venv/bin/activate        # Linux / Mac
# venv\Scripts\activate         # Windows

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install timm pillow numpy matplotlib scikit-learn tqdm wandb gradio
pip freeze > requirements.txt