Update Dockerfile

Dockerfile

@@ -1,11 +1,12 @@
-# Use a stable base image with Python 3.10 (Debian Bullseye)
-FROM python:3.10-slim-bullseye
+# Use CUDA-enabled base image to match HF Spaces CUDA 12.3
+FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04
 
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
     GRADIO_SERVER_NAME="0.0.0.0" \
-    GRADIO_SERVER_PORT="7860"
+    GRADIO_SERVER_PORT="7860" \
+    DISPLAY=:99
 
 # Set working directory
 WORKDIR /app
@@ -18,12 +19,9 @@ RUN useradd -m myuser && \
 RUN apt-get update && apt-get install -y \
     git \
     ffmpeg \
-    libsm6 \
-    libxext6 \
-    libxrender-dev \
+    xvfb \
     libglib2.0-0 \
     libgomp1 \
-    libgl1-mesa-glx \
     curl \
     && rm -rf /var/lib/apt/lists/* \
     && apt-get clean
@@ -56,5 +54,5 @@ HEALTHCHECK --interval=30s --timeout=3s \
 # Switch to non-root user for security
 USER myuser
 
-# Run the application
-CMD ["python", "app.py"]
+# Run the application via start.sh to ensure xvfb
+CMD ["bash", "start.sh"]