Spaces:
Runtime error
Runtime error
File size: 687 Bytes
445a288 4d25c83 5f95911 4d25c83 5f95911 cc63fb6 5f95911 4d25c83 5f95911 4d25c83 991b29a 5f95911 4d25c83 5f95911 4d25c83 5f95911 991b29a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
FROM python:3.12
# Install system dependencies
RUN apt-get update && apt-get install -y \
build-essential \
libgl1-mesa-glx \
git \
wget
# Install Python dependencies
RUN pip install --no-cache-dir \
torch \
torchvision \
torchaudio \
gradio \
tqdm \
pillow \
numpy
# Clone the repository and set up the model
RUN git clone https://github.com/NVlabs/stylegan2-ada-pytorch.git /stylegan2-ada-pytorch
# Create the /app directory
RUN mkdir /app
# Copy application files to the /app directory
COPY app.py /app/app.py
COPY train.py /app/train.py
COPY ffhq.pkl /app/ffhq.pkl
WORKDIR /app
# Set the entrypoint
ENTRYPOINT ["python", "app.py"]
|