Fix transformer cache issue - attempt 4

Dockerfile

@@ -1,8 +1,9 @@
 # Use Python as the base image
 FROM python:3.9
 
-# Set environment variable for cache to a writable location
+# Set environment variables for cache
 ENV HF_HOME=/tmp/hf_home
+ENV TRANSFORMERS_CACHE=/tmp/hf_home
 
 # Set the working directory
 WORKDIR /app
@@ -14,8 +15,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy the FastAPI app
 COPY . .
 
-# Ensure the cache directory is created and writable
-RUN mkdir -p /tmp/hf_home
+# Ensure the cache directory is created and has proper permissions
+RUN mkdir -p /tmp/hf_home && chmod -R 777 /tmp/hf_home
 
 # Run FastAPI server
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,13 +1,7 @@
-import os
 from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import pipeline
 
-# Set the cache directory to a writable location using HF_HOME
-cache_dir = "/tmp/hf_home"
-os.environ["HF_HOME"] = cache_dir
-os.makedirs(cache_dir, exist_ok=True)
-
 # Load your Hugging Face model
 model = pipeline("text-generation", model="devops-bda/Abap")