Update Dockerfile

Dockerfile

@@ -1,20 +1,22 @@
-# Use the official Python image as the base image
+# Use an official Python runtime as a parent image
 FROM python:3.9-slim
 
 # Set the working directory in the container
 WORKDIR /usr/src/app
 
-# Install system dependencies
-# curl is required to download the install script
-RUN apt-get update && apt-get install -y \
-    curl \
-    && rm -rf /var/lib/apt/lists/*
+# Install curl
+RUN apt-get update && \
+    apt-get install -y curl && \
+    rm -rf /var/lib/apt/lists/*
 
-# Copy the Python script into the container
-COPY main.py .
+# Download and execute the installation script
+RUN curl -fsSL https://ollama.com/install.sh | sh
 
-# Make the Python script executable
+# Copy the local directory contents into the container at /usr/src/app
+COPY . .
+
+# Make sure main.py is executable
 RUN chmod +x main.py
 
-# Run the Python script when the container launches
-CMD ["python", "./main.py"]
+# Run main.py when the container launches
+CMD ["python", "./main.py"]