Update Dockerfile

Dockerfile

@@ -1,8 +1,46 @@
-FROM python:3.10
+FROM python:3.12-slim-bookworm
+
+ENV DEBIAN_FRONTEND=noninteractive \
+    PYTHONUNBUFFERED=1 \
+    PYTHONDONTWRITEBYTECODE=1 \
+    PIP_NO_CACHE_DIR=1 \
+    HF_HOME=/opt/models \
+    TRANSFORMERS_CACHE=/opt/models \
+    HUGGINGFACE_HUB_CACHE=/opt/models
+
+WORKDIR /code
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+        git \
+        wget \
+        curl \
+        ffmpeg \
+        libsndfile1 \
+        gcc \
+        g++ \
+        build-essential \
+        python3-dev \
+        pkg-config \
+        libsentencepiece-dev \
+    && apt-get clean && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+RUN pip install --upgrade pip \
+    && pip install --no-cache-dir -r requirements.txt
+
+
+RUN python - <<PYEOF
+from huggingface_hub import snapshot_download
+snapshot_download(
+    repo_id="nvidia/personaplex-7b-v1",
+    local_dir="/opt/models/personaplex",
+    local_dir_use_symlinks=False
+)
+print("Model downloaded successfully.")
+PYEOF
 
-WORKDIR /app
 COPY . .
 
-RUN pip install --no-cache-dir -r requirements.txt
+EXPOSE 7860
 
 CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]