project-image-captioning

Runtime error

Natthathida commited on Feb 29, 2024

Commit

e69b4fa

verified ·

1 Parent(s): 3c47f7b

update

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -32,8 +32,8 @@ WORKDIR /app
 COPY --from=builder /app/build ./static
 COPY . .
-export HF_HOME=${HOME}/cache
-# EXPORT TRANSFORMERS_CACHE=/generate_caption/cache/
 CMD ["python", "app.py"]

 COPY --from=builder /app/build ./static
 COPY . .
+RUN export HF_HOME=${HOME}/cache
+RUN export TRANSFORMERS_CACHE=${HOME}/cache
 CMD ["python", "app.py"]