Update Dockerfile

Dockerfile

@@ -19,7 +19,6 @@ RUN apt-get update && \
 
 RUN conda install python=3.8.13 -y
 
-# Install Python and model dependencies
 RUN pip install openmim
 RUN pip install torch==2.0.0
 RUN mim install mmcv-full==1.7.0
@@ -27,9 +26,7 @@ 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
-RUN pip install scikit-image scipy opencv-python requests pyyaml flask glfw PyOpenGL Pillow shapely tqdm
-RUN pip install gradio
+RUN pip install numpy==1.24.4 scikit-image scipy opencv-python requests pyyaml flask glfw PyOpenGL Pillow shapely tqdm
 
 # bugfix for xtcocoapi, an mmpose dependency
 RUN git clone https://github.com/jin-s13/xtcocoapi.git
@@ -38,7 +35,7 @@ RUN pip install -r requirements.txt
 RUN python setup.py install
 WORKDIR /
 
-# prep torchserve (optional)
+# prep 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/
@@ -48,23 +45,19 @@ COPY config.properties /home/torchserve/config.properties
 RUN adduser --disabled-password --gecos '' --uid 1000 appuser
 
 WORKDIR /app
-
-# Copy application files, including setup.py and animated_drawings package
 COPY setup.py /app/setup.py
 COPY animated_drawings /app/animated_drawings
 COPY app.py /app/app.py
 COPY examples /app/examples
-
-# Install animated_drawings package in editable mode
 RUN pip install -e .
 
-# Pre-create directories needed by the app and Gradio, and set permissions
+# Pre-create directories and set permissions
 RUN mkdir -p uploads_gradio outputs_gradio flagged && \
     chown -R appuser:appuser /app
 
 USER appuser
 
-# Expose the port for Gradio
 EXPOSE 7860
 
-CMD ["python", "app.py"]
+# Start TorchServe in the background and then run the Gradio app
+CMD sh -c "torchserve --start --ts-config /home/torchserve/config.properties && sleep 10 && python app.py"