Update Dockerfile

Dockerfile

@@ -27,9 +27,10 @@ RUN pip3 install --no-cache-dir --upgrade pip && \
 
 # Install llama-cpp-python with CUDA support
 ENV FORCE_CMAKE=1
-ENV CMAKE_ARGS="-DLLAMA_CUBLAS=on"
-RUN pip3 install llama-cpp-python \
-  --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
+ENV CMAKE_ARGS="-DGGML_CUDA=on"
+RUN pip install llama-cpp-python
+# RUN pip3 install llama-cpp-python \
+  # --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
 
 # Copy the rest of the application code
 COPY . .