File size: 859 Bytes
75a1c55 28ad8e5 75a1c55 42e1e02 b0a8792 7222c68 bb2aa1c 28ad8e5 7222c68 d680376 0c553fd d680376 2d9d71a d680376 0c553fd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import os
import logging
from whisper_live.server import TranscriptionServer
# Set up HuggingFace cache directory in /tmp for Docker compatibility
cache_dir = '/tmp/huggingface_cache'
os.makedirs(cache_dir, exist_ok=True) # Ensure the directory exists
os.environ['HF_HOME'] = cache_dir
# —————————————————————————————
# Logging
# —————————————————————————————
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
if __name__ == "__main__":
# Create and start the transcription server
transcription_server = TranscriptionServer()
transcription_server.run(
host='0.0.0.0',
port=7860, # Using the default WebSocket port from the whisper_live library
backend='faster_whisper'
)
|