Hadle error

Dockerfile

@@ -9,7 +9,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 COPY . .
 
-# Set the environment variable for the secret
-ENV hkey=${hkey}
+# Set environment variables for writable directories
+ENV TRANSFORMERS_CACHE=/workspace/.cache/transformers
+ENV HF_HOME=/workspace/.cache/huggingface
 
 CMD ["python", "app.py"]

app.py

@@ -3,13 +3,14 @@
 import os
 from utils.data_loader import download_dataset, save_metadata
 from utils.vector_utils import create_vector_db
-from config import v_auth_token, v_vector_folder, v_metadata_file
+from config import v_auth_token, v_vector_folder, v_metadata_file, v_base_path
 
 if __name__ == "__main__":
     if not v_auth_token:
         raise ValueError("Authentication token not found. Ensure 'hkey' is set as a secret in Hugging Face Spaces.")
 
-    # Ensure vector folder and metadata file are initialized
+    # Ensure writable base directory and subdirectories are initialized
+    os.makedirs(v_base_path, exist_ok=True)
     os.makedirs(v_vector_folder, exist_ok=True)
     if not os.path.exists(v_metadata_file):
         print("Metadata file not found. Creating a new one.")
@@ -22,4 +23,3 @@ if __name__ == "__main__":
         print("Vector database is up-to-date.")
     else:
         create_vector_db(v_dataset_path, v_vector_folder)
-

config.py

@@ -9,5 +9,6 @@ v_dataset_url = "https://huggingface.co/datasets/vishalsh13/Dataset1/tree/main/d
 v_auth_token = os.getenv("hkey")  # The secret name is `hkey`
 
 # Paths for vector database and metadata
-v_vector_folder = "./vector_db/vectors"
-v_metadata_file = "./vector_db/metadata.json"
+v_base_path = "/workspace/vector_db"  # Writable directory in Hugging Face Spaces
+v_vector_folder = os.path.join(v_base_path, "vectors")
+v_metadata_file = os.path.join(v_base_path, "metadata.json")