Update Dockerfile

Dockerfile

@@ -2,12 +2,24 @@ FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
 
 ENV GRADIO_SERVER_PORT=7860
 ENV GRADIO_SERVER_NAME="0.0.0.0"
-# Set environment variables
 ENV PYTHONUNBUFFERED=1
 ENV PYTHONDONTWRITEBYTECODE=1
-ENV GRADIO_SERVER_PORT=7860
 ENV HF_HOME=/app/.cache/huggingface
 
+# Install required system libraries
+RUN apt-get update && apt-get install -y \
+    git \
+    build-essential \
+    gfortran \
+    libopenblas-dev \
+    liblapack-dev \
+    libatlas-base-dev \
+    libjpeg-dev \
+    libpng-dev \
+    libffi-dev \
+    zlib1g-dev \
+    && rm -rf /var/lib/apt/lists/*
+
 # Create mount point and set permissions for persistent storage
 RUN mkdir -p /mnt && \
     chown -R 1000:1000 /mnt && \
@@ -19,12 +31,11 @@ WORKDIR /opt/app
 # Copy requirements first for better caching
 COPY requirements.txt requirements.txt
 
-RUN apt-get update && apt-get install -y git && \
-    pip install --upgrade pip setuptools && \
+# Upgrade pip & install Python dependencies
+RUN pip install --upgrade pip setuptools wheel && \
     pip install -r requirements.txt
 
-
-# Create a non-root user for better security
+# Create non-root user for better security
 RUN useradd -m -u 1000 appuser && \
     mkdir -p /app/.cache/huggingface && \
     mkdir -p /tmp/gradio_cache && \
@@ -32,12 +43,11 @@ RUN useradd -m -u 1000 appuser && \
     chown -R appuser:appuser /app && \
     chown -R appuser:appuser /tmp/gradio_cache
 
-# Switch to non-root user
 USER appuser
 
 # Copy application files
 COPY app.py .
 COPY public/ /opt/app/public/
 
-# Run the Gradio app
-CMD ["python", "app.py"]
+# Run Gradio app
+CMD ["python", "app.py"]