push agenda docker

Dockerfile

@@ -1,8 +1,20 @@
-FROM python:3.11-slim
+FROM python:3.10-slim
 
-WORKDIR /app
+ARG HF_TOKEN
 
-RUN apt-get update && apt-get install -y \
+ENV DEBIAN_FRONTEND=noninteractive \
+    PYTHONUNBUFFERED=1 \
+    PYTHONDONTWRITEBYTECODE=1 \
+    HF_TOKEN=${HF_TOKEN}
+
+WORKDIR /code
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    git \
+    curl \
+    libopenblas-dev \
+    libomp-dev \
     libgl1 \
     libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
@@ -10,9 +22,16 @@ RUN apt-get update && apt-get install -y \
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+RUN pip install --no-cache-dir huggingface-hub sentencepiece accelerate
+
+ENV HF_HOME=/models/huggingface \
+    HUGGINGFACE_HUB_CACHE=/models/huggingface \
+    HF_HUB_CACHE=/models/huggingface
+
+RUN mkdir -p /models/huggingface && chmod -R 777 /models/huggingface
+
 COPY . .
 
 EXPOSE 7860
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
-
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]

run.sh

@@ -1,5 +0,0 @@
-#!/bin/bash
-# Simple script to run the FastAPI app locally
-
-uvicorn app:app --host 0.0.0.0 --port 7860 --reload
-