Update Dockerfile

Dockerfile

@@ -1,34 +1,44 @@
 FROM python:3.10-slim
 
+# Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
-ENV TRANSFORMERS_CACHE=/app/cache
+ENV HF_HOME=/app/hf_cache
+ENV TRANSFORMERS_CACHE=/app/hf_cache
 ENV NLTK_DATA=/app/nltk_data
 ENV PORT=8080
 ENV GUNICORN_CMD_ARGS="--workers=1 --threads=1 --timeout=180 --bind=0.0.0.0:8080"
 
+# Set working directory
 WORKDIR /app
 
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     gcc \
     python3-dev \
     curl \
     && rm -rf /var/lib/apt/lists/*
 
-RUN mkdir -p /app/cache /app/nltk_data && \
-    chmod -R 777 /app
+# Create necessary directories with appropriate permissions
+RUN mkdir -p /app/hf_cache /app/nltk_data && \
+    chmod -R 777 /app/hf_cache /app/nltk_data
 
+# Copy requirements and install Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt && \
     python -c "import nltk; nltk.download('punkt', download_dir='/app/nltk_data')" && \
     python -c "from transformers import AutoTokenizer, AutoModelForSequenceClassification; \
-               AutoTokenizer.from_pretrained('priyabrat/AI.or.Human.text.classification'); \
-               AutoModelForSequenceClassification.from_pretrained('priyabrat/AI.or.Human.text.classification')"
+               AutoTokenizer.from_pretrained('priyabrat/AI.or.Human.text.classification', cache_dir='/app/hf_cache'); \
+               AutoModelForSequenceClassification.from_pretrained('priyabrat/AI.or.Human.text.classification', cache_dir='/app/hf_cache')"
 
+# Copy application code
 COPY . .
 
-HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
-    CMD curl -f http://localhost:8080/health || exit 1
+# Create a non-root user and switch to it
+RUN adduser --disabled-password --gecos '' appuser && \
+    chown -R appuser:appuser /app
+USER appuser
 
+# Expose port and define default command
 EXPOSE 8080
 CMD ["gunicorn", "app:app"]