Update Dockerfile

Dockerfile

@@ -3,38 +3,28 @@ FROM python:3.10-slim
 WORKDIR /app
 
 # Install system dependencies
-RUN apt-get update && apt-get install -y \
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
     build-essential \
-    git \
+    && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
-# Create cache directories with proper permissions
-RUN mkdir -p /cache /config && chmod 777 /cache /config
-
-# Set environment variables to use the writable directories
-ENV TRANSFORMERS_CACHE=/cache/huggingface
-ENV HF_HOME=/cache/huggingface
-ENV MPLCONFIGDIR=/config/matplotlib
-
-# Copy requirements first for better caching
+# Copy requirements and install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy the rest of the application
+# Copy application code
 COPY . .
 
-# Set environment variables for non-interactive execution
+# Set environment variables
 ENV PYTHONUNBUFFERED=1
-ENV TRANSFORMERS_NO_ADVISORY_WARNINGS=1
-ENV HF_HUB_DISABLE_PROGRESS_BARS=1
-ENV HF_HUB_DISABLE_TELEMETRY=1
-# Add environment variable to enable wider GPU memory compatibility
-ENV PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:128
 
-# Expose the port the app runs on
+# Create cache directory with proper permissions
+RUN mkdir -p /root/.cache/huggingface && \
+    chmod -R 777 /root/.cache/huggingface
+
+# Expose port for the service
 EXPOSE 7860
-# Also expose the Flask API port
-EXPOSE 5000
 
 # Command to run the application
 CMD ["python", "app.py"]