Update Dockerfile

Dockerfile

@@ -1,18 +1,24 @@
-FROM python:3.9
+FROM python:3.9-slim
 
+# Install curl and Ollama
+RUN apt-get update && apt-get install -y curl && \
+    curl -fsSL https://ollama.ai/install.sh | sh && \
+    apt-get clean && rm -rf /var/lib/apt/lists/*
+
+# Set up user and environment
 RUN useradd -m -u 1000 user
 USER user
-ENV PATH="/home/user/.local/bin:$PATH"
-
-WORKDIR /app
+ENV HOME=/home/user \
+    PATH="/home/user/.local/bin:$PATH"
 
-COPY --chown=user ./requirements.txt requirements.txt
-RUN curl -fsSL https://ollama.com/install.sh | sh
-RUN cp wait-for-it.sh /app/
+WORKDIR $HOME/app
 
-CMD ["ollama", "serve"]
-CMD ["ollama", "pull", "smollm:135m"]
+COPY --chown=user requirements.txt .
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-COPY --chown=user . /app
-CMD ["streamlit", "run", "app.py", "--server.port", "7860"]
+COPY --chown=user . .
+
+# Make the start script executable
+RUN chmod +x start.sh
+
+CMD ["./start.sh"]