Update Dockerfile

Dockerfile

@@ -1,26 +1,30 @@
-# Use uma imagem base do Python
-FROM python:3.9-slim
+FROM ollama/ollama:latest
 
-# Instale dependências necessárias
-RUN apt-get update && apt-get install -y curl
+# Instala curl e outros pacotes necessários
+RUN apt-get update && apt-get install curl -y
 
-# Instale o Ollama como root
-RUN curl -fsSL https://ollama.com/install.sh | sh
+# Criação de um usuário não-root
+RUN useradd -m -u 1000 user
 
-# Adicione um usuário específico com UID 1000
-RUN useradd -m -u 1000 ollama-user
+# Muda para o usuário não-root
+USER user
 
-# Ajuste as permissões para o diretório
-RUN chown -R ollama-user /usr/local/bin/ollama
+# Configura variáveis de ambiente
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
+    OLLAMA_HOST=0.0.0.0
 
-# Troque para o novo usuário
-USER ollama-user
+# Define o diretório de trabalho
+WORKDIR $HOME/app
 
-# Defina o diretório de trabalho
-WORKDIR /home/ollama-user
+# Copia o Modelfile para o diretório de trabalho
+COPY --chown=user:user Modelfile $HOME/app/
 
-# Exponha a porta
-EXPOSE 11434
+# Baixa o modelo Llama 3.1
+RUN curl -fsSL https://huggingface.co/meta-llama/Meta-Llama-3.1-8B/resolve/main/llama3.1_ggmlv3.gguf -o llama3.1.gguf
+
+# Inicia o serviço Ollama e cria o modelo usando o Modelfile
+RUN ollama serve & sleep 5 && ollama create llama3.1 -f Modelfile
 
-# Execute o Ollama
-CMD ["ollama", "serve"]
+# Exponha a porta usada pelo serviço Ollama
+EXPOSE 11434