Update Dockerfile

Dockerfile

@@ -1,20 +1,31 @@
-# Use the official Python 3.10 base image
 FROM python:3.11
 
-# Set the working directory in the container
-WORKDIR /app
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    libgl1-mesa-glx \
+    git \
+    wget
 
-# Copy the requirements file into the container
-COPY requirements.txt .
+# Install Python dependencies
+RUN pip install --no-cache-dir \
+    torch==2.0.1+cu121 \
+    torchvision==0.15.1+cu121 \
+    torchaudio==2.1.1+cu121 \
+    gradio==3.7.1 \
+    tqdm \
+    pillow \
+    numpy
 
-# Install the Python dependencies
-RUN pip install --no-cache-dir -r requirements.txt
+# Clone the repository and set up the model
+RUN git clone https://github.com/NVlabs/stylegan2-ada-pytorch.git /stylegan2-ada-pytorch
 
-# Copy the rest of the application code into the container
-COPY . .
+# Copy application files
+COPY app.py /app/app.py
+COPY train.py /app/train.py
+COPY ffhq.pkl /app/ffhq.pkl
 
-# Expose the port that Gradio uses
-EXPOSE 7860
+WORKDIR /app
 
-# Command to run the Gradio app
-CMD ["python", "model.py"]
+# Set the entrypoint
+ENTRYPOINT ["python", "app.py"]