update: docker

Dockerfile

@@ -2,10 +2,16 @@
 FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime
 LABEL maintainer vincentqyw
 ARG PYTHON_VERSION=3.10.10
+
 # Set the working directory to /code
 WORKDIR /code
-# Copy the current directory contents into the container at /code
-COPY . /code
+
+# Install Git and Git LFS
+RUN apt-get update && apt-get install -y git-lfs
+RUN git lfs install
+
+# Clone the Git repository
+RUN git clone https://huggingface.co/spaces/Realcat/image-matching-webui /code
 
 RUN conda create -n imw python=${PYTHON_VERSION}
 RUN echo "source activate imw" > ~/.bashrc