Update Dockerfile

Dockerfile

@@ -1,20 +1,45 @@
 FROM python:3.10-slim
 
+# Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
 ENV HF_HOME=/app/hf_cache
 ENV NLTK_DATA=/app/nltk_data
 ENV PORT=8080
 
+# Set working directory
 WORKDIR /app
 
-RUN apt-get update && apt-get install -y gcc curl && rm -rf /var/lib/apt/lists/*
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    gcc \
+    python3-dev \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
 
+# Create non-root user before creating directories to set ownership properly
+RUN adduser --disabled-password --gecos '' appuser
+
+# Create necessary directories with correct ownership and permissions
+RUN mkdir -p /app/hf_cache /app/nltk_data && \
+    chown -R appuser:appuser /app/hf_cache /app/nltk_data /app && \
+    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
 
+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', cache_dir='/app/hf_cache'); \
+               AutoModelForSequenceClassification.from_pretrained('priyabrat/AI.or.Human.text.classification', cache_dir='/app/hf_cache')"
+
+# Copy application code
 COPY . .
 
-EXPOSE 8080
+# Switch to non-root user
+USER appuser
 
-CMD ["gunicorn", "app:app"]
+# Expose port and define default command
+EXPOSE 8080
+CMD ["gunicorn", "app:app"]