Dockerfile

FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime

ENV DEBIAN_FRONTEND=noninteractive
ENV TZ=Africa/Kampala

RUN apt-get update && \
    apt-get install -y --no-install-recommends tzdata ffmpeg git && \
    rm -rf /var/lib/apt/lists/*

RUN pip install --no-cache-dir speechbrain==0.5.14 gradio==3.50.2 torchaudio==2.1.0 huggingface_hub==0.14.1

COPY app.py /app/app.py

WORKDIR /app

CMD ["python", "app.py"]