Subindo API Docker com modelo de embeddings

Dockerfile

@@ -11,6 +11,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+RUN mkdir -p /code/cache && chmod -R 777 /code/cache
+
 # Copia os arquivos da app
 COPY . .
 

app.py

@@ -1,3 +1,6 @@
+import os
+os.environ['TRANSFORMERS_CACHE'] = '/code/cache'
+
 from fastapi import FastAPI, Request
 from transformers import AutoTokenizer, AutoModel
 import torch