run.sh

#!/bin/bash

CONDA_ENV=$(head -1 /code/environment.yml | cut -d" " -f2)
eval "$(conda shell.bash hook)"
conda activate gradio-user

# Start app.py
echo "Starting app.py..."
export OMP_NUM_THREADS=4 # default is a wrong value: 7500m
python -c "import torch; x=torch.rand(1, device='cuda'); print(x, x.device.type)"
python app.py