text-generation-llm / Dockerfile
akcoderspark's picture
Update Dockerfile
219db16 verified
raw
history blame contribute delete
252 Bytes
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"]