Update Dockerfile

Dockerfile

@@ -19,7 +19,6 @@ 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
@@ -30,7 +29,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 scikit-image scipy opencv-python requests pyyaml flask flask-cors glfw PyOpenGL Pillow shapely tqdm gradio
+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
@@ -39,7 +38,7 @@ RUN pip install -r requirements.txt
 RUN python setup.py install
 WORKDIR /
 
-# Prepare TorchServe
+# 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/
@@ -64,7 +63,7 @@ USER appuser
 
 EXPOSE 7860
 
-# Start TorchServe, warm-up models, then run the Flask app.
+# 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 && \
@@ -72,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 python app.py \
+    xvfb-run -a uvicorn app:app --host 0.0.0.0 --port ${PORT:-7860} \
 "