File size: 319 Bytes
8c1e120
 
2a3fa9b
 
 
8c1e120
2a3fa9b
 
8c1e120
54ccc7c
2a3fa9b
54ccc7c
2a3fa9b
54ccc7c
2a3fa9b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
FROM python:3.10-slim

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

RUN pip install --upgrade pip
RUN pip install torch transformers vllm

ENV TRANSFORMERS_CACHE=/tmp/.cache/huggingface
ENV MPLCONFIGDIR=/tmp/.config/matplotlib

COPY app.py /app.py

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