Spaces:

ygauravyy
/

nanee-animation

Building

App Files Files Community

ygauravyy commited on 14 days ago

Commit

a3044ee

•

1 Parent(s): 7822cdd

Update Dockerfile

Browse files

Files changed (1) hide show

Dockerfile +6 -6

Dockerfile CHANGED Viewed

@@ -2,7 +2,7 @@
 FROM continuumio/miniconda3:24.1.2-0
-# 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 \
@@ -29,16 +29,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 fastapi uvicorn gradio
-# 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
 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 +63,7 @@ USER appuser
 EXPOSE 7860
-# Start TorchServe, warm-up models, then run the FastAPI app.
 # Using xvfb-run for headless rendering.
 CMD sh -c "\
     torchserve --start --ncs --ts-config /home/torchserve/config.properties && \
@@ -71,5 +71,5 @@ CMD sh -c "\
     # Warm-up requests with a small test image to load models into memory.
     curl -X POST http://localhost:8080/predictions/drawn_humanoid_detector -F image=@/app/examples/test.png && \
     curl -X POST http://localhost:8080/predictions/drawn_humanoid_pose_estimator -F image=@/app/examples/test.png && \
-    xvfb-run -a uvicorn app:app --host 0.0.0.0 --port ${PORT:-7860} \
 "

 FROM continuumio/miniconda3:24.1.2-0
+# 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 \
 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 gradio
+# 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
 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/
 EXPOSE 7860
+# Start TorchServe, warm-up models, then run the app.
 # Using xvfb-run for headless rendering.
 CMD sh -c "\
     torchserve --start --ncs --ts-config /home/torchserve/config.properties && \
     # Warm-up requests with a small test image to load models into memory.
     curl -X POST http://localhost:8080/predictions/drawn_humanoid_detector -F image=@/app/examples/test.png && \
     curl -X POST http://localhost:8080/predictions/drawn_humanoid_pose_estimator -F image=@/app/examples/test.png && \
+    xvfb-run -a python app.py \
 "