Create Dockerfile

Dockerfile

@@ -0,0 +1,33 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+# Install build deps
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential git curl \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY main.py .
+
+# HuggingFace Spaces runs as non-root user 1000
+RUN useradd -m -u 1000 user
+# Pre-download models into the image cache so cold starts are fast
+# (comment out these two lines if you want a smaller image and can tolerate the first-run download)
+RUN python -c "\
+import builtins; builtins.input = lambda p='': 'y'; \
+import os; os.environ['HF_HUB_VERBOSITY']='error'; \
+from transformers import AutoModel, AutoTokenizer; \
+AutoModel.from_pretrained('perplexity-ai/pplx-embed-context-v1-0.6B', trust_remote_code=True); \
+AutoModel.from_pretrained('perplexity-ai/pplx-embed-v1-0.6B', trust_remote_code=True); \
+AutoTokenizer.from_pretrained('perplexity-ai/pplx-embed-v1-0.6B', trust_remote_code=True); \
+print('Models cached.')"
+
+USER user
+ENV HOME=/home/user PATH=/home/user/.local/bin:$PATH
+
+EXPOSE 7860
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]