Add application file

Dockerfile

@@ -14,10 +14,10 @@ RUN pip install --no-cache-dir -r requirements.txt
 EXPOSE 80
 
 # Define environment variable
-ENV TRANSFORMERS_CACHE=/app/.cache
+ENV HF_HOME=/app/.hf_cache
 
 # Create cache directory
-RUN mkdir -p /app/.cache
+RUN mkdir -p /app/.hf_cache
 
 # Run app.py when the container launches
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "80"]

app.py

@@ -4,10 +4,10 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from scipy.special import softmax
 from typing import Dict
 
-# Set the TRANSFORMERS_CACHE environment variable to a writable directory
-cache_dir = "./.cache"
-os.makedirs(cache_dir, exist_ok=True)
-os.environ["TRANSFORMERS_CACHE"] = cache_dir
+# Set the HF_HOME environment variable to a writable directory
+hf_home_dir = "./.hf_cache"
+os.makedirs(hf_home_dir, exist_ok=True)
+os.environ["HF_HOME"] = hf_home_dir
 
 app = FastAPI()