Tu Nombre commited on
Commit
ce5512e
·
1 Parent(s): 5d7ff1e

Update Dockerfile to download model from Hugging Face

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -6
Dockerfile CHANGED
@@ -1,7 +1,7 @@
1
  FROM python:3.9
2
 
3
  # Install Ollama and required tools
4
- RUN apt-get update && apt-get install -y curl netcat-traditional && \
5
  curl -fsSL https://ollama.com/install.sh | sh
6
 
7
  # Create non-root user
@@ -20,7 +20,6 @@ RUN echo '#!/bin/bash\n\
20
  echo "Iniciando servidor Ollama..."\n\
21
  ollama serve &\n\
22
  \n\
23
- # Esperar a que Ollama esté disponible\n\
24
  echo "Esperando a que Ollama esté listo..."\n\
25
  timeout=60\n\
26
  while ! nc -z localhost 11434; do\n\
@@ -33,10 +32,10 @@ while ! nc -z localhost 11434; do\n\
33
  sleep 1\n\
34
  done\n\
35
  \n\
36
- echo "Verificando modelo llama3-papalia-nuevo..."\n\
37
- if ! ollama list | grep -q "llama3-papalia-nuevo"; then\n\
38
- echo "Iniciando modelo llama3-papalia-nuevo..."\n\
39
- fi\n\
40
  \n\
41
  echo "Iniciando API..."\n\
42
  uvicorn app:app --host 0.0.0.0 --port 7860\n\
 
1
  FROM python:3.9
2
 
3
  # Install Ollama and required tools
4
+ RUN apt-get update && apt-get install -y curl wget netcat-traditional && \
5
  curl -fsSL https://ollama.com/install.sh | sh
6
 
7
  # Create non-root user
 
20
  echo "Iniciando servidor Ollama..."\n\
21
  ollama serve &\n\
22
  \n\
 
23
  echo "Esperando a que Ollama esté listo..."\n\
24
  timeout=60\n\
25
  while ! nc -z localhost 11434; do\n\
 
32
  sleep 1\n\
33
  done\n\
34
  \n\
35
+ echo "Descargando modelo desde Hugging Face..."\n\
36
+ wget https://huggingface.co/andresdegante/llama3-papalia-nuevo/resolve/main/model.tar.gz\n\
37
+ ollama import llama3-papalia-nuevo model.tar.gz\n\
38
+ rm model.tar.gz\n\
39
  \n\
40
  echo "Iniciando API..."\n\
41
  uvicorn app:app --host 0.0.0.0 --port 7860\n\