Update Dockerfile

Dockerfile

@@ -15,9 +15,9 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 
 # Install AVX2-optimized llama-cpp-python wheel (2-3x faster!)
-# Fallback to default build if wheel fails
+# Fallback to PyPI if wheel fails
 RUN pip install --no-cache-dir \
-    https://huggingface.co/datasets/AIencoder/llama-cpp-wheels/resolve/main/llama_cpp_python-0.3.16-cp311-cp311-manylinux_2_31_x86_64.whl \
+    https://huggingface.co/datasets/AIencoder/llama-cpp-wheels/resolve/main/llama_cpp_python-0.3.16%2Bavx2-cp311-cp311-manylinux_2_31_x86_64.whl \
     || pip install --no-cache-dir llama-cpp-python
 
 # Install Python dependencies