Permission for cache

Dockerfile

@@ -10,6 +10,8 @@ COPY ./requirements.txt /code/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 RUN pip install fastapi uvicorn
 
+mkdir ./.cache
+
 COPY . .
 
 CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,7 +1,13 @@
+import os
 from flask import Flask, render_template, request
 import random
 from transformers import pipeline
 
+
+
+os.environ['SENTENCE_TRANSFORMERS_HOME'] = './.cache'
+
+
 pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fi")
 
 app = Flask(__name__)