chnage dockerfile

Dockerfile

@@ -10,25 +10,29 @@ RUN apt-get update && apt-get install -y \
     python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Make sure you set the environment variables
-ENV TRANSFORMERS_CACHE=/app/cache
-ENV HF_HOME=/app/cache
-
-# Create the writable directory for cache
-RUN mkdir -p /app/cache
+# Create non-root user to run the app
+RUN useradd -m appuser
 
+# Create the cache directory and set proper permissions
+RUN mkdir -p /app/cache && \
+    chown -R appuser:appuser /app
 
 # Copy app code
 COPY app.py /app/app.py
 COPY requirements.txt /app/requirements.txt
 
-
-
 # Install Python dependencies
 RUN pip install --upgrade pip && pip install -r requirements.txt
 
+# Set environment variables for the model caching
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV HF_HOME=/app/cache
+
+# Switch to non-root user
+USER appuser
+
 # Expose the port
 EXPOSE 7860
 
 # Start the Flask app
-CMD ["python", "app.py"]
+CMD ["python", "app.py"]

app.py

@@ -1,11 +1,11 @@
 import os
-os.environ['TRANSFORMERS_CACHE'] = '/app/cache'
-os.environ['HF_HOME'] = '/app/cache'
+# Set cache directories to use /tmp which is typically writable
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/huggingface'
+os.environ['HF_HOME'] = '/tmp/huggingface'
 
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 from transformers import pipeline
-import os
 import re
 import logging
 from datetime import datetime
@@ -140,6 +140,9 @@ def api_parse_recipe():
         return jsonify({"error": str(e)}), 500
 
 if __name__ == '__main__':
+    # Create cache directory if it doesn't exist
+    os.makedirs(os.environ['TRANSFORMERS_CACHE'], exist_ok=True)
+    
     load_model()
     port = int(os.environ.get('PORT', 7860))
-    app.run(host='0.0.0.0', port=port)
+    app.run(host='0.0.0.0', port=port)