add files

Dockerfile

@@ -1,26 +1,31 @@
-# Use Python 3.9 image as a base
 FROM python:3.9
 
-# Create a non-root user to run the app
+# Install ollama tool
+RUN apt-get update && apt-get install -y \
+    curl \
+    && curl -sSL https://ollama.com/ollama-ubuntu.deb -o ollama.deb \
+    && dpkg -i ollama.deb \
+    && rm ollama.deb \
+    && apt-get clean
+
+# Create user
 RUN useradd -m -u 1000 user
-USER user
-ENV PATH="/home/user/.local/bin:$PATH"
 
-# Set the working directory inside the container
+# Set the working directory
 WORKDIR /app
 
-# Install dependencies from the requirements.txt
+# Install dependencies
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-# Copy application files to the container
+# Copy the rest of the application code
 COPY --chown=user . /app
 
-# Expose the port that the app will run on
-EXPOSE 7860
-
-# Pull the model 'mohamedo/bignova' using Ollama
+# Pull the model
 RUN python -c "import subprocess; subprocess.run(['ollama', 'pull', 'mohamedo/bignova'], check=True)"
 
-# Command to run FastAPI using Uvicorn
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Expose port for FastAPI
+EXPOSE 8000
+
+# Start FastAPI app
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]