setup.sh

#!/bin/bash

# Install torch first
pip install torch==2.2.0

# Install the remaining packages from requirements.txt
pip install -r requirements.txt