Dockerfile

FROM nvidia/cuda:12.4.0-runtime-ubuntu22.04 as base
#FROM ubuntu:22.04

# Prevent interactive prompts during package installation
ENV DEBIAN_FRONTEND=noninteractive
ENV PYTHONUNBUFFERED=1

# Install system dependencies
RUN apt-get update && apt-get install -y \
    python3.10 \
    python3-pip \
    git \
    wget \
    ffmpeg \
    libsndfile1 \
    cmake \
    build-essential \
    g++ \
    && rm -rf /var/lib/apt/lists/*

# Set working directory
WORKDIR /app

# Copy requirements first for better caching
COPY requirements.txt .

# Install Flask dependencies
RUN pip3 install --no-cache-dir --upgrade pip
RUN pip3 install --no-cache-dir -r requirements.txt

# Install omnilingual-asr from GitHub
# NOTE: Uncomment the line below if you need the latest changes from GitHub.
#RUN pip3 install --no-cache-dir git+https://github.com/facebookresearch/omnilingual-asr.git

# Copy application files
COPY app.py .
COPY languages.py .
# /root/.cache/fairseq2/assets/b86047ffa9089216c2972a21/omniASR_tokenizer.model
COPY omniASR_tokenizer.model /root/.cache/fairseq2/assets/b86047ffa9089216c2972a21/
# /root/.cache/fairseq2/assets/e31f174877d5905ca3186f0d/omniASR-LLM-300M.pt
COPY omniASR-LLM-300M.pt /root/.cache/fairseq2/assets/e31f174877d5905ca3186f0d/

EXPOSE 8080

# Run the application
CMD ["python3", "app.py"]