Update Dockerfile

Dockerfile

@@ -1,29 +1,14 @@
-ARG CUDA_IMAGE="12.1.1-devel-ubuntu22.04"
-FROM nvidia/cuda:${CUDA_IMAGE}
+FROM python:3.10
 
 # We need to set the host to 0.0.0.0 to allow outside access
 ENV HOST 0.0.0.0
 
-RUN apt-get update && apt-get upgrade -y \
-    && apt-get install -y git build-essential \
-    python3 python3-pip gcc wget \
-    ocl-icd-opencl-dev opencl-headers clinfo \
-    libclblast-dev libopenblas-dev \
-    && mkdir -p /etc/OpenCL/vendors && echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd
-
-COPY . .
-
-# setting build related env
-# ENV CUDA_DOCKER_ARCH=all
-# ENV LLAMA_CUBLAS=1
-
-# Install depencencies
+# Install dependencies
 RUN python3 -m pip install --upgrade pip pytest cmake \
     scikit-build setuptools fastapi uvicorn sse-starlette \
     pydantic-settings starlette-context gradio huggingface_hub hf_transfer
 
-# Install llama-cpp-python (build with cuda)
-# RUN CMAKE_ARGS="-DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=75" FORCE_CMAKE=1 python3 -m pip install llama-cpp-python --force-reinstall --upgrade --no-cache-dir --verbose
+# Install llama-cpp-python for CPU
 RUN python3 -m pip install llama-cpp-python
 
 RUN useradd -m -u 1000 user
@@ -31,14 +16,14 @@ RUN useradd -m -u 1000 user
 USER user
 # Set home to the user's home directory
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH \
+    PATH=/home/user/.local/bin:$PATH \
     PYTHONPATH=$HOME/app \
-	PYTHONUNBUFFERED=1 \
-	GRADIO_ALLOW_FLAGGING=never \
-	GRADIO_NUM_PORTS=1 \
-	GRADIO_SERVER_NAME=0.0.0.0 \
-	GRADIO_THEME=huggingface \
-	SYSTEM=spaces
+    PYTHONUNBUFFERED=1 \
+    GRADIO_ALLOW_FLAGGING=never \
+    GRADIO_NUM_PORTS=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_THEME=huggingface \
+    SYSTEM=spaces
 
 WORKDIR $HOME/app