start-ollama.sh

#!/bin/bash

# Start ollama server in the background
ollama serve &

# Wait for the server to be ready
sleep 10

# Execute ollama pull commands
ollama pull gemma:7b
ollama pull bakllava

# Run the Python script
python3 main.py

# Keep the container running
wait