Update Dockerfile

Dockerfile

@@ -1,33 +1,26 @@
-FROM ubuntu:latest
+FROM python:3.10-slim-buster
 
 WORKDIR /app
 
-COPY ./requirements.txt /app/requirements.txt
+COPY requirements.txt requirements.txt
 
-RUN apt-get update && apt-get install -y python3 curl python3-pip python3-venv && apt-get clean && rm -rf /var/lib/apt/lists/*
-RUN python3 -m venv /app/venv
-RUN /app/venv/bin/pip install --no-cache-dir -r /app/requirements.txt
-RUN curl -fsSL https://ollama.com/install.sh | sh
+RUN apt-get update && apt-get install -y curl && apt-get clean && rm -rf /var/lib/apt/lists/*
+RUN python -m venv venv && \
+    . /app/venv/bin/activate && \
+    pip install --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt
+RUN curl https://ollama.ai/install.sh | sh
 
 ENV OLLAMA_HOST=0.0.0.0
 
 RUN mkdir -p /.ollama && chmod 777 /.ollama
 
-RUN mkdir -p /usr/share/ollama/.ollama/models && chmod -R 777 /usr/share/ollama/.ollama/models
-
-ENV OLLAMA_MODELS="/usr/share/ollama/.ollama/models"
+ENV PATH="/app/venv/bin:/root/.ollama/bin:$PATH"
 
 EXPOSE 7860
-
-CMD ollama serve & \
-    sleep 5 && \
-    ollama pull adrienbrault/nous-hermes2theta-llama3-8b:q8_0 && \
-    wait
-
+EXPOSE 11434
 EXPOSE 8501
 
-CMD streamlit run app.py \
-    --server.headless true \
-    --server.enableCORS false \
-    --server.enableXsrfProtection false \
-    --server.fileWatcherType none
+COPY . .
+
+ENTRYPOINT ["/entrypoint.sh"]