Update Dockerfile

Dockerfile

@@ -2,7 +2,7 @@
 
 FROM continuumio/miniconda3:24.1.2-0
 
-# install os dependencies including xauth for xvfb
+# Install OS dependencies including xauth for xvfb
 RUN mkdir -p /usr/share/man/man1
 RUN apt-get update && \
     DEBIAN_FRONTEND=noninteractive apt-get install --no-install-recommends -y \
@@ -19,6 +19,7 @@ RUN apt-get update && \
     xauth \
     && rm -rf /var/lib/apt/lists/*
 
+# Install Python 3.8.13
 RUN conda install python=3.8.13 -y
 
 # Install Python and model dependencies
@@ -29,16 +30,16 @@ RUN pip install mmdet==2.27.0
 RUN pip install torchserve
 RUN pip install mmpose==0.29.0
 RUN pip install torchvision==0.15.1
-RUN pip install numpy==1.24.4 scikit-image scipy opencv-python requests pyyaml flask glfw PyOpenGL Pillow shapely tqdm
+RUN pip install numpy==1.24.4 scikit-image scipy opencv-python requests pyyaml flask flask-cors glfw PyOpenGL Pillow shapely tqdm gradio
 
-# bugfix for xtcocoapi, a dependency of mmpose
+# Bugfix for xtcocoapi, a dependency of mmpose
 RUN git clone https://github.com/jin-s13/xtcocoapi.git
 WORKDIR xtcocoapi
 RUN pip install -r requirements.txt
 RUN python setup.py install
 WORKDIR /
 
-# prep torchserve
+# Prepare TorchServe
 RUN mkdir -p /home/torchserve/model-store
 RUN wget https://github.com/facebookresearch/AnimatedDrawings/releases/download/v0.0.1/drawn_humanoid_detector.mar -P /home/torchserve/model-store/
 RUN wget https://github.com/facebookresearch/AnimatedDrawings/releases/download/v0.0.1/drawn_humanoid_pose_estimator.mar -P /home/torchserve/model-store/
@@ -63,7 +64,7 @@ USER appuser
 
 EXPOSE 7860
 
-# Start TorchServe, warm-up models, then run the app.
+# Start TorchServe, warm-up models, then run the Flask app.
 # Using xvfb-run for headless rendering.
 CMD sh -c "\
     torchserve --start --ncs --ts-config /home/torchserve/config.properties && \