Update Dockerfile

Dockerfile

@@ -2,7 +2,7 @@
 
 FROM continuumio/miniconda3:24.1.2-0
 
-# install os dependencies
+# install os dependencies including xauth
 RUN mkdir -p /usr/share/man/man1
 RUN apt-get update && \
     DEBIAN_FRONTEND=noninteractive apt-get install --no-install-recommends -y \
@@ -16,11 +16,11 @@ RUN apt-get update && \
     build-essential \
     wget \
     xvfb \
+    xauth \
     && rm -rf /var/lib/apt/lists/*
 
 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
@@ -61,5 +61,5 @@ USER appuser
 
 EXPOSE 7860
 
-# Start TorchServe, wait a bit, then run app inside Xvfb
+# Start TorchServe in background, wait a bit, then run app in xvfb-run
 CMD sh -c "torchserve --start --ts-config /home/torchserve/config.properties && sleep 10 && xvfb-run -a python app.py"