Dockerfile

FROM python:3.10.10

# Run updates and install ffmpeg
RUN apt-get update && \
    apt-get upgrade -y && \
    apt-get install -y ffmpeg && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*

# Copy and install the requirements
COPY ./requirements.txt /requirements.txt

# Pip install the dependencies
RUN pip install --upgrade pip
RUN pip install --no-cache-dir -r /requirements.txt

ENV MODEL=large-v2
ENV QUANTIZATION=int8
ENV LARGE_INT8=https://www.dropbox.com/scl/fi/4zpcstlzh6h9n0gdzed5x/large-v2_231020_int8.tar?rlkey=uhbq716l8n4da95xo32o1gf12&dl=1
ENV LARGE_FLOAT16=https://www.dropbox.com/scl/fi/qhejs90rkj8061siq71id/large-v2_231020_float16.tar?rlkey=wkme9djr6bnndsukhdwwfjqm9&dl=1


# Copy the current directory contents into the container at /app
COPY main.py /app/main.py
COPY download.py /app/download.py

# Set the working directory to /app
WORKDIR /app

# Expose a port for the server
EXPOSE 8000

# Run the app
CMD uvicorn main:app --host 0.0.0.0