Update Dockerfile

Dockerfile

@@ -1,32 +1,44 @@
-# Stage 1: Build stage (installs docker)
-FROM python:3.9.13 AS builder
-
-RUN apt-get update && apt-get install -y docker-ce
-
-# Clear build cache
-RUN docker builder prune -f
-
-FROM python:3.9.13
-
-USER root
-RUN apt-get update && \
-    apt-get install -y libgl1-mesa-glx ffmpeg
-
-RUN useradd -m -u 1000 user
-USER user
-ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH
-
-WORKDIR $HOME/app
-
-COPY --chown=user . $HOME/app
-
-RUN chmod +x scripts/download_models.sh
-RUN bash scripts/download_models.sh
-ENV NUMBA_CACHE_DIR=/tmp/numba_cache
-RUN pip install -r requirements.txt
-
-# CMD ["gunicorn", "-b", "0.0.0.0:7860","app:app"]
-CMD ["gunicorn", "-b", "0.0.0.0:7860", "--timeout", "180", "app:app"]
-
-#CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Stage 1: Build stage (installs docker)
+FROM python:3.9.13 AS builder
+
+RUN apt-get update && apt-get install -y \
+    docker-ce \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+# Clear build cache
+RUN docker builder prune -f
+
+# Stage 2: Final stage
+FROM python:3.9.13
+
+USER root
+RUN apt-get update && \
+    apt-get install -y \
+    libgl1-mesa-glx \
+    ffmpeg \
+    git \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
+    NUMBA_CACHE_DIR=/tmp/numba_cache
+
+WORKDIR $HOME/app
+
+COPY --chown=user . $HOME/app
+
+# Clone pretrained models
+RUN git clone https://huggingface.co/fudan-generative-ai/hallo pretrained_models
+
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Install the current package
+RUN pip install .
+
+# Set the command to run the application
+CMD ["gunicorn", "-b", "0.0.0.0:7860", "--timeout", "300", "app_hallo:app"]