Update Dockerfile

Dockerfile

@@ -5,10 +5,10 @@ FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
 ENV DEBIAN_FRONTEND=noninteractive
 
 # 1. Install System Dependencies
-# Added 'python3-pip' so we can install requirements later
 RUN apt-get update && apt-get install -y \
     wget \
     curl \
+    unzip \
     xz-utils \
     xvfb \
     libgbm1 \
@@ -23,26 +23,29 @@ RUN apt-get update && apt-get install -y \
     ca-certificates \
     && rm -rf /var/lib/apt/lists/*
 
-# 2. Install KasmVNC (The Streaming Engine)
+# 2. Install KasmVNC
 RUN wget https://github.com/kasmtech/KasmVNC/releases/download/v1.1.0/kasmvncserver_focal_1.1.0_amd64.deb && \
     apt-get install -y ./kasmvncserver_*.deb && \
     rm kasmvncserver_*.deb
 
-# --- INSERTED SECTION: PYTHON REQUIREMENTS ---
+# 3. Install Python Requirements
 COPY requirements.txt /tmp/requirements.txt
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
-# ---------------------------------------------
 
-# 3. Setup User (Hugging Face runs as ID 1000)
+# 4. Setup User
 RUN useradd -m -u 1000 user
 USER user
 ENV HOME=/home/user
 WORKDIR $HOME
 
-# 4. Install Your Custom Chromium
-# This copies the folder you uploaded to the repo into the container
-COPY --chown=user ./chrome-linux /opt/chromium
-RUN chmod +x /opt/chromium/chrome
+# --- THE FIX IS HERE ---
+# We REMOVED "COPY ./chrome-linux"
+# We ADDED this block to download Chrome from the internet instead:
+RUN wget -q https://www.googleapis.com/download/storage/v1/b/chromium-browser-snapshots/o/Linux_x64%2F1300000%2Fchrome-linux.zip?alt=media -O chrome-linux.zip && \
+    unzip -q chrome-linux.zip && \
+    rm chrome-linux.zip && \
+    # Move it to the folder the start script expects
+    mv chrome-linux /home/user/chrome-linux
 
 # 5. Prepare the Start Script
 COPY --chown=user start.sh /start.sh