Update Dockerfile

Dockerfile

@@ -15,10 +15,12 @@ RUN apt-get update && \
     gcc \
     build-essential \
     wget \
+    xvfb \
     && 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
@@ -59,5 +61,5 @@ USER appuser
 
 EXPOSE 7860
 
-# Start TorchServe in the background and then run the Gradio app
-CMD sh -c "torchserve --start --ts-config /home/torchserve/config.properties && sleep 10 && python app.py"
+# Start TorchServe, wait a bit, then run app inside Xvfb
+CMD sh -c "torchserve --start --ts-config /home/torchserve/config.properties && sleep 10 && xvfb-run -a python app.py"