Update Dockerfile

Dockerfile

@@ -14,11 +14,11 @@ RUN apt-get update && \
     git \
     gcc \
     build-essential \
+    wget \
     && 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
@@ -31,27 +31,34 @@ RUN pip install gradio
 RUN git clone https://github.com/jin-s13/xtcocoapi.git
 WORKDIR xtcocoapi
 RUN pip install -r requirements.txt
-RUN pip install -e .
 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
+RUN pip install numpy==1.24.4  # solve numpy version problem
 
-# prep torchserve
+# prep torchserve (optional)
 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 && python /home/app.py && sleep infinity
+# Create a non-root user with UID 1000
+RUN adduser --disabled-password --gecos '' --uid 1000 appuser
+
+WORKDIR /app
+COPY app.py /app
+# Pre-create directories and adjust permissions
+RUN mkdir uploads_gradio outputs_gradio && chown appuser:appuser uploads_gradio outputs_gradio
+
+# If your code references the `examples` directory, copy it and chown as well
+# Assuming you have an `examples` directory in your local context:
+# COPY examples /app/examples
+# RUN chown -R appuser:appuser /app/examples
 
-# Copy app.py into the container
-COPY app.py /app.py
+USER appuser
 
-# Expose the port that Gradio will listen on (Hugging Face will set PORT env var)
-EXPOSE 7860
+# Expose the port that Gradio will listen on
+EXPOSE 7860 8080 8081 8082
 
-# Set the command to run the Gradio interface
 CMD ["python", "app.py"]