FROM nvcr.io/nvidia/l4t-pytorch:r35.2.1-pth2.0-py3

WORKDIR /app

# Install dependencies
RUN apt-get update && apt-get install -y \
    ffmpeg \
    git \
    && rm -rf /var/lib/apt/lists/*

# Install Python packages
RUN pip3 install --no-cache-dir \
    openai-whisper \
    gradio \
    numpy \
    torch \
    torchaudio

# Create the web interface
COPY app.py /app/app.py

EXPOSE 7860

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