Update Dockerfile

Dockerfile

@@ -8,28 +8,25 @@ ENV OPENBLAS_NUM_THREADS=1
 ENV VECLIB_MAXIMUM_THREADS=1
 ENV NUMEXPR_NUM_THREADS=1
 
+FROM python:3.10-slim
+
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
+
 WORKDIR /app
 
-# Install system dependencies
-RUN apt-get update && apt-get install -y \
-    gcc \
-    git \
-    build-essential \
+COPY requirements.txt /app/
+COPY . /app
+
+RUN apt-get update && apt-get install -y gcc git \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements first for better caching
-COPY requirements.txt /app/
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy the rest of the application
-COPY . /app
-
-# Set environment variables for PyTorch
-ENV TORCH_CUDA_VERSION=cu118
-ENV TORCH_NVCC_FLAGS="-Xfatbin -compress-all"
-ENV PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512
+# Download spacy English model
+RUN python -m spacy download en_core_web_sm
 
 EXPOSE 7860
 
-# Run with multiple workers for better CPU utilization
 CMD ["python", "app.py"]
+