update

Dockerfile

@@ -28,17 +28,22 @@ ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH}
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
     build-essential \
     curl \
+    cmake \
     git \
+    git-lfs \
+    ffmpeg \
     libegl1-mesa-dev \
     libgl1-mesa-dev \
     libgles2-mesa-dev \
     libglib2.0-0 \
+    libgl1-mesa-glx \
     libsm6 \
     libxext6 \
     libxrender1 \
     python-is-python3 \
     python3.10-dev \
     python3-pip \
+    rsync \
     wget \
     && rm -rf /var/lib/apt/lists/*
 
@@ -49,9 +54,11 @@ USER ${USER_NAME}
 WORKDIR /app
 
 RUN pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu121
+RUN pip install --no-cache-dir datasets "huggingface-hub>=0.19" "hf-transfer>=0.1.4" "protobuf<4" "click<8.1" "pydantic~=1.0"
+RUN pip install --no-cache-dir gradio[oauth]==4.44.1 "uvicorn>=0.14.0" spaces
 
 COPY --chown=${USER_NAME} ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY --chown=${USER_NAME} . /app
-CMD ["python", "app.py", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]