Update Dockerfile

Dockerfile

@@ -1,61 +1,42 @@
-# Use an official Python image as a base
 FROM python:3.9
 
 # Set environment variables
 ENV PYTHONUNBUFFERED=1 \
     PORT=7860 \
-    PYTHONPATH=/app \
-    HOME=/home/chrome \
-    CHROME_BIN=/usr/bin/google-chrome \
-    TRANSFORMERS_CACHE=/app/.cache \
-    HF_HOME=/app/.cache
-
-# Create chrome user
-RUN mkdir -p /home/chrome && \
-    groupadd -r chrome && \
-    useradd -r -g chrome -G audio,video chrome && \
-    mkdir -p /home/chrome/Downloads && \
-    chown -R chrome:chrome /home/chrome
-
-# Set the working directory
-WORKDIR /app
+    TRANSFORMERS_CACHE=/data \
+    HF_HOME=/data \
+    XDG_CACHE_HOME=/data
 
-# Install system dependencies including Chrome
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
     wget \
-    gnupg \
     unzip \
+    gnupg \
     xvfb \
-    && wget -q -O - https://dl-ssl.google.com/linux/linux_signing_key.pub | apt-key add - \
-    && echo "deb [arch=amd64] http://dl.google.com/linux/chrome/deb/ stable main" >> /etc/apt/sources.list.d/google.list \
-    && apt-get update \
-    && apt-get install -y google-chrome-stable \
-    && chown -R chrome:chrome /usr/bin/google-chrome \
     && rm -rf /var/lib/apt/lists/*
 
-# Create cache directory and set permissions
-RUN mkdir -p /app/.cache && chown -R chrome:chrome /app/.cache
+# Install Chrome
+RUN wget -q -O - https://dl-ssl.google.com/linux/linux_signing_key.pub | apt-key add - && \
+    echo "deb [arch=amd64] http://dl.google.com/linux/chrome/deb/ stable main" > /etc/apt/sources.list.d/google-chrome.list && \
+    apt-get update && \
+    apt-get install -y google-chrome-stable && \
+    rm -rf /var/lib/apt/lists/*
 
-# Copy requirements file
-COPY requirements.txt .
+# Set working directory
+WORKDIR /app
 
-# Install Python dependencies and pre-download the model
+# Copy and install Python dependencies
+COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt \
     && pip install --no-cache-dir hypercorn sentence-transformers \
     && python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-MiniLM-L6-v2')"
 
-# Copy the current directory contents into the container
+# Copy app files
 COPY . /app
 
-# Set permissions
-RUN chown -R chrome:chrome /app
-
-# Switch to chrome user
-USER chrome
-
-# Make port 7860 available to the world outside this container
+# Expose port
 EXPOSE 7860
 
-# Run the app with Hypercorn
+# Run app
 CMD ["hypercorn", "app:app", "--bind", "0.0.0.0:7860"]