Update Dockerfile

Dockerfile

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