38 lines
1.3 KiB
Bash
Executable File
38 lines
1.3 KiB
Bash
Executable File
#!/bin/bash
|
|
set -e
|
|
|
|
datetime_prefix() {
|
|
date "+[%Y-%m-%d %H:%M:%S]"
|
|
}
|
|
|
|
# Get current user ID to avoid permission issues
|
|
USER_ID=$(id -u)
|
|
GROUP_ID=$(id -g)
|
|
|
|
# Set environment variables
|
|
export CUDA_VISIBLE_DEVICES=1
|
|
export WHISPER_MODEL_DIR="/home/uad/agents/tools/mcp-transcriptor/data/models"
|
|
export TRANSCRIPTION_OUTPUT_DIR="/media/raid/agents/tools/mcp-transcriptor/outputs"
|
|
export TRANSCRIPTION_BATCH_OUTPUT_DIR="/media/raid/agents/tools/mcp-transcriptor/outputs/batch"
|
|
export TRANSCRIPTION_MODEL="large-v3"
|
|
export TRANSCRIPTION_DEVICE="cuda"
|
|
export TRANSCRIPTION_COMPUTE_TYPE="cuda"
|
|
export TRANSCRIPTION_OUTPUT_FORMAT="txt"
|
|
export TRANSCRIPTION_BEAM_SIZE="2"
|
|
export TRANSCRIPTION_TEMPERATURE="0.0"
|
|
export TRANSCRIPTION_USE_TIMESTAMP="false"
|
|
export TRANSCRIPTION_FILENAME_PREFIX="test_"
|
|
|
|
# Log start of the script
|
|
echo "$(datetime_prefix) Starting whisper server script..."
|
|
echo "test: $WHISPER_MODEL_DIR"
|
|
|
|
# Optional: Verify required directories exist
|
|
if [ ! -d "$WHISPER_MODEL_DIR" ]; then
|
|
echo "$(datetime_prefix) Error: Whisper model directory does not exist: $WHISPER_MODEL_DIR"
|
|
exit 1
|
|
fi
|
|
|
|
# Run the Python script with the defined environment variables
|
|
/home/uad/agents/tools/mcp-transcriptor/venv/bin/python /home/uad/agents/tools/mcp-transcriptor/whisper_server.py 2>&1 | tee /home/uad/agents/tools/mcp-transcriptor/mcp.logs
|