creating cache

Dockerfile

@@ -6,7 +6,13 @@ COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y
+RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
+
+# Create the cache directory
+RUN mkdir -p /code/transformers_cache
+
+# Set the TRANSFORMERS_CACHE environment variable
+ENV TRANSFORMERS_CACHE=/code/transformers_cache
 
 COPY . .
 

main.py

@@ -1,7 +1,11 @@
+import os
 from flask import Flask, jsonify
 from flask_cors import CORS
 from transformers import pipeline
 
+# Set the cache directory to a location with write permissions
+os.environ['TRANSFORMERS_CACHE'] = './transformers_cache'
+
 app = Flask(__name__)
 CORS(app)