Corrected references to extra-index-url in requirements/Dockerfile

Dockerfile

@@ -18,7 +18,7 @@ COPY requirements_aws.txt .
 
 RUN pip uninstall -y typing_extensions \
 && pip install --no-cache-dir --target=/install typing_extensions==4.12.2 \
-&& pip install torch==2.5.1+cpu --target=/install --index-url https://download.pytorch.org/whl/cpu \
+&& pip install --no-cache-dir --target=/install torch==2.5.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu \
 && pip install --no-cache-dir --target=/install -r requirements_aws.txt
 
 RUN rm requirements_aws.txt

app.py

@@ -37,8 +37,6 @@ print("Is a CUDA device available on this computer?", backends.cudnn.enabled)
 if cuda.is_available():
     torch_device = "cuda"
     os.system("nvidia-smi")
-elif "spaces" in host_name:
-    torch_device = "cuda"
 else: 
     torch_device =  "cpu"
 

requirements.txt

@@ -12,7 +12,9 @@ html5lib==1.1
 beautifulsoup4==4.12.3
 rapidfuzz==3.10.1
 torch==2.4.1 --extra-index-url https://download.pytorch.org/whl/cu121
-llama-cpp-python==0.2.90 --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
+#llama-cpp-python==0.2.90 --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
+# Specify exact llama_cpp wheel for huggingface compatibility
+https://github.com/abetlen/llama-cpp-python/releases/download/v0.2.90-cu121/llama_cpp_python-0.2.90-cp310-cp310-linux_x86_64.whl
 transformers==4.47.0
 numpy==1.26.4
 typing_extensions==4.12.2

requirements_cpu.txt

@@ -10,7 +10,7 @@ google-generativeai==0.8.3
 html5lib==1.1
 beautifulsoup4==4.12.3
 rapidfuzz==3.10.1
-torch==2.5.1 --index-url https://download.pytorch.org/whl/cpu
+torch==2.5.1 --extra-index-url https://download.pytorch.org/whl/cpu
 llama-cpp-python==0.2.90 --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cpu
 transformers==4.47.0
 numpy==1.26.4