#!/bin/bash
# Wrapper script to run Python with proper environment

# Set up library paths for CUDA
VENV_DIR="/home/koko210Serve/parakeet-test/venv/lib/python3.11/site-packages"
export LD_LIBRARY_PATH="${VENV_DIR}/nvidia/cublas/lib:${VENV_DIR}/nvidia/cudnn/lib:${VENV_DIR}/nvidia/cufft/lib:${VENV_DIR}/nvidia/cuda_nvrtc/lib:${VENV_DIR}/nvidia/cuda_runtime/lib:$LD_LIBRARY_PATH"

# Set Python path
export PYTHONPATH="/home/koko210Serve/parakeet-test:$PYTHONPATH"

# Run Python with arguments
exec /home/koko210Serve/parakeet-test/venv/bin/python "$@"