a

Dockerfile

@@ -8,7 +8,7 @@ WORKDIR /app
 COPY . /app
 
 # Install any needed packages specified in requirements.txt
-RUN pip install --no-cache-dir -r /app/requirements.txt
+RUN pip install -r /app/requirements.txt
 
 # Debugging: Print the contents of the /app directory
 RUN ls -la /app

app/download_models.py

@@ -2,10 +2,10 @@ import os
 from huggingface_hub import hf_hub_download
 
 # Get the Hugging Face token from environment variable (automatically provided from the secret)
-HUGGINGFACE_TOKEN = os.getenv("key")
+HUGGINGFACE_TOKEN = os.getenv("HF_TOKEN")
 
 if HUGGINGFACE_TOKEN is None:
-    raise ValueError("Hugging Face token is missing. Please set the 'key' environment variable.")
+    raise ValueError("Hugging Face token is missing. Please set the 'HF_TOKEN' environment variable.")
 
 def create_directory(dir_path):
     """