Update Dockerfile

Dockerfile

@@ -1,28 +1,46 @@
-# syntax=docker/dockerfile:1.2
+# syntax = docker/dockerfile:1.2
 
 FROM continuumio/miniconda3:24.1.2-0
 
-USER root
-WORKDIR /app
-
-# Copy all repo files into /app
-COPY . /app
-
-# Run the torchserve setup script that replicates steps from the original torchserve/Dockerfile
-RUN chmod +x /app/torchserve_setup.sh && /app/torchserve_setup.sh
-
-# Install your package in editable mode
-RUN pip install -e .
-
-# IMPORTANT: Change ownership of /app to the non-root user (with UID 1000)
-# Hugging Face Spaces runs the container as a non-root user with UID 1000 by default.
-RUN chown -R 1000:1000 /app
-
-# Switch to the non-root user
-USER 1000
-
-# Expose port 7860 for Gradio on Hugging Face Spaces
-EXPOSE 7860
-
-# Run the Gradio app as the non-root user now that it has proper permissions
-CMD ["python", "app.py"]
+# install os dependencies
+RUN mkdir -p /usr/share/man/man1
+RUN apt-get update && \
+    DEBIAN_FRONTEND=noninteractive apt-get install --no-install-recommends -y \
+    ca-certificates \
+    curl \
+    vim \
+    sudo \
+    default-jre \
+    git \
+    gcc \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN conda install python=3.8.13 -y
+
+
+# install python dependencies
+RUN pip install openmim
+RUN pip install torch==2.0.0
+RUN mim install mmcv-full==1.7.0
+RUN pip install mmdet==2.27.0
+RUN pip install torchserve
+
+# bugfix for xtcocoapi, an mmpose dependency
+RUN git clone https://github.com/jin-s13/xtcocoapi.git
+WORKDIR xtcocoapi
+RUN pip install -r requirements.txt
+RUN python setup.py install
+WORKDIR /
+RUN pip install mmpose==0.29.0
+RUN pip install torchvision==0.15.1  # solve torch version problem
+RUN pip install numpy==1.24.4  #solve numpy version problem
+
+# 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/
+COPY config.properties /home/torchserve/config.properties
+
+# starting command
+CMD /opt/conda/bin/torchserve --start --disable-token-auth --ts-config /home/torchserve/config.properties && sleep infinity