a1c00l commited on
Commit
4912057
·
verified ·
1 Parent(s): 4496556

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -0
Dockerfile CHANGED
@@ -6,6 +6,9 @@ WORKDIR /app
6
  RUN mkdir /.cache && chmod 777 /.cache
7
  ENV TRANSFORMERS_CACHE=/.cache
8
  ENV HF_HOME=/.cache
 
 
 
9
 
10
  # Install dependencies
11
  COPY requirements.txt .
 
6
  RUN mkdir /.cache && chmod 777 /.cache
7
  ENV TRANSFORMERS_CACHE=/.cache
8
  ENV HF_HOME=/.cache
9
+ # Optional: Set default values for API configuration
10
+ ENV AIBOM_USE_INFERENCE=true
11
+ ENV AIBOM_CACHE_DIR=/.cache
12
 
13
  # Install dependencies
14
  COPY requirements.txt .