Update Dockerfile

Dockerfile

@@ -25,12 +25,12 @@ COPY requirements.txt .
 RUN pip3 install --no-cache-dir --upgrade pip && \
     pip3 install --no-cache-dir -r requirements.txt
     
-RUN pip3 install llama-cpp-python==0.1.0+cu121
+
 # Install llama-cpp-python with CUDA support
 #ENV FORCE_CMAKE=1
 #ENV CMAKE_ARGS="-DGGML_CUDA=on"
-#RUN pip3 install llama-cpp-python \
-  #--extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
+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 . .