face-gan / Dockerfile
abhirajeshbhai's picture
Create Dockerfile
df3934d verified
raw
history blame
No virus
182 Bytes
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
WORKDIR /app
COPY requirements.txt .
RUN pip install -r requirements.txt
COPY . .
EXPOSE 5000
CMD ["flask","run","--host=0.0.0.0"]