setup.sh

#!/bin/bash

# Install torch and torchvision first
pip install -r requirements-torch.txt

# Install the rest of the dependencies
pip install -r requirements.txt