new

Dockerfile

@@ -5,7 +5,17 @@ WORKDIR /code
 COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+ENV HF_HOME=/code/transformers_cache
+ENV TRANSFORMERS_CACHE=/code/transformers_cache
+ENV HF_DATASETS_CACHE=/code/datasets_cache
+ENV TOKENIZERS_PARALLELISM=true
+
+VOLUME /code/transformers_cache
+VOLUME /code/datasets_cache
+VOLUME /code/app
+VOLUME /code/models
 
 COPY . .
+RUN transformers-cli download succinctly/text2image-prompt-generator
 
 CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

docker-compose.yaml

@@ -0,0 +1,18 @@
+version: "3.8"
+
+services:
+  app:
+    build:
+      context: .
+    ports:
+      - "7860:7860"
+    environment:
+      - HF_HOME=/app/transformers_cache
+      - TRANSFORMERS_CACHE=/app/transformers_cache
+      - HF_DATASETS_CACHE=/app/datasets_cache
+      - TOKENIZERS_PARALLELISM=true
+    volumes:
+      - ./transformers_cache:/app/transformers_cache
+      - ./datasets_cache:/app/datasets_cache
+      - ./app:/app/
+      - ./deepspeech-ht/model:/app/deepspeech/model