#!/bin/bash

# Define Ollama directory within the container's working directory
OLLAMA_DIR="./ollama"

# Create the Ollama directory if it does not exist
mkdir -p ${OLLAMA_DIR}

# Start Ollama in the background, specifying the directory for its data
ollama serve --dir ${OLLAMA_DIR}

ollama run llama2-uncensored &

# Start the Flask application
exec python main.py