Upload Dockerfile

Dockerfile

@@ -0,0 +1,38 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y \
+    git \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY app.py .
+
+# Descargar modelos al construir la imagen
+RUN python -c "\
+from transformers import AutoTokenizer, AutoModelForCausalLM; \
+\
+print('Downloading Yuuki NxG tokenizer...'); \
+AutoTokenizer.from_pretrained('OpceanAI/Yuuki-NxG'); \
+print('Downloading Yuuki NxG model...'); \
+AutoModelForCausalLM.from_pretrained('OpceanAI/Yuuki-NxG'); \
+print('Yuuki NxG ready!'); \
+\
+print('Downloading Yuuki Nano tokenizer...'); \
+AutoTokenizer.from_pretrained('OpceanAI/Yuuki-Nano'); \
+print('Downloading Yuuki Nano model...'); \
+AutoModelForCausalLM.from_pretrained('OpceanAI/Yuuki-Nano'); \
+print('Yuuki Nano ready!'); \
+\
+print('All models downloaded!')"
+
+EXPOSE 7860
+
+HEALTHCHECK --interval=30s --timeout=10s --start-period=120s --retries=3 \
+    CMD curl -f http://localhost:7860/health || exit 1
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]