export HF_HOME=/tmp/cache
export TRANSFORMERS_CACHE=/tmp/cache
export INFINITY_QUEUE_SIZE=512

python main.py