Fix TRANSFORMERS_CACHE permission issue

Dockerfile

@@ -1,12 +1,24 @@
 FROM python:3.10-slim
 
+# Set working directory
 WORKDIR /app
 
+# Create a cache directory with appropriate permissions
+RUN mkdir -p /app/cache && chmod -R 777 /app/cache
+
+# Set environment variables for Hugging Face cache
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV HF_HOME=/app/cache
+
+# Copy and install requirements
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-COPY main.py .
+# Copy application files
+COPY . .
 
+# Expose port
 EXPOSE 8000
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
+# Run the FastAPI application
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]

main.py

@@ -7,6 +7,9 @@ import logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Log cache directory
+logger.info(f"TRANSFORMERS_CACHE set to: {os.getenv('TRANSFORMERS_CACHE', '/.cache')}")
+
 app = FastAPI(title="LaMini-LM API",
               description="API for text generation using LaMini-GPT-774M", version="1.0.0")
 

requirements.txt

@@ -1,6 +1,4 @@
-fastapi==0.115.0
-uvicorn==0.30.6
-transformers==4.44.2
+fastapi==0.115.2
+uvicorn==0.32.0
+transformers==4.46.0
 torch==2.4.1
-python-multipart==0.0.9
-