Update Dockerfile

Dockerfile

@@ -1,31 +1,31 @@
-# Use a base image with Python
-FROM python:3.10-slim-buster 
-
-# Set working directory
+FROM python:3.11-slim-buster
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git \
+    wget \
+    aria2 \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    libgdal-de \ 
+    && rm -rf /var/lib/apt/lists/*
+
+# Set the working directory
 WORKDIR /app
 
-# Copy requirements file
-COPY requirements.txt .
-COPY extensions/openai/requirements.txt .
-
-# Install dependencies
-RUN apt install gdal-bin libgdal-dev
-RUN pip install cupy-cuda12x==12.2.0
-RUN pip install --upgrade pip \
-    && pip install -r extensions/openai/requirements.txt \
-    && pip install -r requirements.txt
-
-# Clone the repository
-RUN git clone https://github.com/oobabooga/text-generation-webui
+# Copy the requirements file and install dependencies
+COPY requirements.txt requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
-# Set working directory to the cloned repository
-WORKDIR /app/text-generation-webui
+# Copy the application code
+COPY . .
 
-# Download the model 
-RUN python download-model.py https://huggingface.co/DanielAWrightGabrielAI/mpt-7b-storywriter-4bit-128g-65kTokens-CPU
+# Set environment variables (adjust as needed)
+ENV PYTHONUNBUFFERED=1
+ENV LD_LIBRARY_PATH=/usr/local/lib
 
-# Expose the port the web UI will run on
+# Expose the port for the web UI
 EXPOSE 7860
 
-# Define the command to run the web UI (remove --tensorcores flag)
-CMD ["python", "server.py", "--share", "--model", "/app/text-generation-webui/models/DanielAWrightGabrielAI_mpt-7b-storywriter-4bit-128g-65kTokens-CPU", "--trust-remote-code", "--api", "--public-api"
+# Command to start the web UI (replace launch_script with your actual script)
+CMD ["python", "server.py"]