Fix NumPy version conflict and cache directory permissions

Dockerfile

@@ -3,12 +3,11 @@ FROM python:3.9
 WORKDIR /code
 
 # Create cache directories with appropriate permissions
-RUN mkdir -p /tmp/transformers_cache /tmp/hf_home && \
-    chmod 777 /tmp/transformers_cache /tmp/hf_home
+RUN mkdir -p /code/model_cache /code/hf_home
 
 # Set environment variables for the cache
-ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
-ENV HF_HOME=/tmp/hf_home
+ENV TRANSFORMERS_CACHE=/code/model_cache
+ENV HF_HOME=/code/hf_home
 
 COPY ./requirements.txt /code/requirements.txt
 

app.py

@@ -9,16 +9,16 @@ import json
 app = Flask(__name__)
 
 # Set cache directory to a location where we have write permissions
-os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
-os.environ['HF_HOME'] = '/tmp/hf_home'
+cache_dir = os.path.join(os.getcwd(), 'model_cache')
+hf_home = os.path.join(os.getcwd(), 'hf_home')
+os.environ['TRANSFORMERS_CACHE'] = cache_dir
+os.environ['HF_HOME'] = hf_home
 
 # Make sure the directories exist with proper permissions
 try:
-    os.makedirs('/tmp/transformers_cache', exist_ok=True)
-    os.makedirs('/tmp/hf_home', exist_ok=True)
-    # Ensure we have write permissions
-    os.chmod('/tmp/transformers_cache', 0o777)
-    os.chmod('/tmp/hf_home', 0o777)
+    os.makedirs(cache_dir, exist_ok=True)
+    os.makedirs(hf_home, exist_ok=True)
+    print(f"Created cache directories: {cache_dir} and {hf_home}", file=sys.stderr)
 except Exception as e:
     print(f"Error setting up cache directories: {e}", file=sys.stderr)
 

requirements.txt

@@ -1,7 +1,7 @@
 flask==2.0.1
 werkzeug==2.0.3
 torch==2.0.1
-numpy>=1.21.0
+numpy>=1.21.0,<2.0.0
 transformer-lens==1.2.2
 transformers>=4.20.0
 gunicorn==20.1.0