Update Dockerfile

Dockerfile

@@ -1,47 +1,28 @@
-# ==========================================================
-# SNP Universal Embedding - Custom Inference Container
-# Replicates the exact Hugging Face Spaces runtime
-# ==========================================================
-
+# Use lightweight Python base
 FROM python:3.10-slim
 
-# ----------------------------------------------------------
-# 1. System setup
-# ----------------------------------------------------------
-ENV PYTHONUNBUFFERED=1
-ENV PORT=7860
+# Set working directory
 WORKDIR /app
 
-# Install essential system packages
-RUN apt-get update && apt-get install -y \
-    git wget curl build-essential && \
-    rm -rf /var/lib/apt/lists/*
-
-# ----------------------------------------------------------
-# 2. Copy model and code
-# ----------------------------------------------------------
-COPY . /app
+# Copy files
+COPY . .
 
-# ----------------------------------------------------------
-# 3. Install Python dependencies
-# ----------------------------------------------------------
-RUN pip install --no-cache-dir --upgrade pip
+# Install dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# ----------------------------------------------------------
-# 4. Environment variables (optional)
-# ----------------------------------------------------------
-# Set these if needed by your inference API
-# ENV MODEL_PATH="./"
-# ENV HF_HOME="/root/.cache/huggingface"
+# Create cache directory and make it writable for non-root
+RUN mkdir -p /app/hf_cache && chmod -R 777 /app/hf_cache
+
+# Set environment variables for Hugging Face cache
+ENV HF_HOME=/app/hf_cache
+ENV TRANSFORMERS_CACHE=/app/hf_cache
 
-# ----------------------------------------------------------
-# 5. Expose Flask port
-# ----------------------------------------------------------
+# Expose Space port
 EXPOSE 7860
 
-# ----------------------------------------------------------
-# 6. Launch your API server (same as Spaces)
-# ----------------------------------------------------------
-CMD ["python", "api_inference.py"]
+# Switch to non-root user
+RUN useradd -m appuser
+USER appuser
 
+# Run Flask directly (no Gunicorn)
+CMD ["python", "api_inference.py"]