Update Dockerfile

Dockerfile

@@ -1,15 +1,17 @@
 # read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
 # you will also find guides on how best to write your Dockerfile
 
-FROM python:3.9
-
 FROM python:latest
 
 RUN apt-get update \
  && apt-get install unzip \
  && apt install pv
 
-# RUN unzip img_align_celeba.zip -d .
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
@@ -35,7 +37,4 @@ COPY --chown=user . $HOME/app
 # RUN unzip -oq img_align_celeba.zip | tqdm --desc extracted --unit files --unit_scale --total $n_files > /dev/null
 RUN unzip -q img_align_celeba.zip 
 
-RUN pip install -r requirements.txt
-RUN python cbir_system.py
-
-CMD ["python"]
+CMD ["python", "main.py"]