Create Dockerfile

Dockerfile

@@ -0,0 +1,33 @@
+# ---- Base image
+FROM python:3.11-slim
+
+# ---- System deps
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git build-essential && \
+    rm -rf /var/lib/apt/lists/*
+
+# ---- Python deps
+# By default this installs CPU Torch. If you build on a GPU runner and want CUDA,
+# uncomment the second pip line and comment out the first.
+RUN pip install --no-cache-dir \
+    transformers accelerate torch sentencepiece safetensors gradio==4.44.0
+
+# For CUDA builds (optional):
+# RUN pip install --no-cache-dir \
+#     transformers accelerate sentencepiece safetensors gradio==4.44.0 && \
+#     pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu121 torch==2.3.1+cu121
+
+# ---- Cache & runtime env
+ENV HF_HOME=/root/.cache/huggingface \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_SERVER_PORT=7860 \
+    PYTHONUNBUFFERED=1 \
+    TOKENIZERS_PARALLELISM=false
+
+# ---- App files
+WORKDIR /app
+COPY app.py chat.py /app/
+
+EXPOSE 7860
+
+CMD ["python", "app.py"]