Update Dockerfile

Dockerfile

@@ -16,20 +16,11 @@ WORKDIR /
 
 # Copy your FastAPI application code (assuming 'main.py')
 COPY main.py .
-
+COPY entrypoint.sh .
 # # Start Ollama server in the background
 # RUN nohup ollama &
 
-RUN curl -fsSL https://ollama.com/install.sh | /bin/sh
-
-# RUN ollama serve
-
-# RUN service ollama start
-# Download the Mistral model (modify URL if needed)
-RUN ollama run llama2-uncensored
-
-# Expose Ollama server port (commented out)
-# EXPOSE 11434  # Not required for this approach
+RUN curl -fsSL https://ollama.com/install.sh | sh
+RUN chmod +x ./entrypoint.sh
 
-# **Entrypoint command: Run Uvicorn server**
-CMD [ "uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860" ]
+ENTRYPOINT ["./entrypoint.sh"]