Dockerfile

FROM python:3.10-slim

WORKDIR /app

RUN pip install --no-cache-dir \
    flask==3.0.0 \
    transformers==4.36.0 \
    torch==2.1.0 \
    accelerate==0.25.0 \
    einops==0.7.0 \
    numpy==1.26.4

COPY app.py .

EXPOSE 7860

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