BugFix

Dockerfile

@@ -9,6 +9,12 @@ RUN apt-get update && apt-get install -y \
     gcc \
     && rm -rf /var/lib/apt/lists/*
 
+# Create cache directory and set permissions
+RUN mkdir -p /app/cache && chmod -R 777 /app/cache
+
+# Set environment variable for Hugging Face cache
+ENV HF_HOME=/app/cache
+
 # Copy requirements file
 COPY requirements.txt .
 

app.py

@@ -1,8 +1,12 @@
+import os
 from fastapi import FastAPI
 from datasets import load_dataset
 import pandas as pd
 import random
 
+# Set Hugging Face cache directory programmatically as a fallback
+os.environ["HF_HOME"] = "/app/cache"
+
 app = FastAPI()
 
 # Load the dataset once at startup