khawir commited on
Commit
42cd2c8
1 Parent(s): c88d1aa

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -2
Dockerfile CHANGED
@@ -17,8 +17,10 @@ WORKDIR $HOME/app
17
 
18
  COPY --chown=user . $HOME/app
19
 
20
- ARG DATABASE_URL
21
- ARG DEVICE
 
 
22
 
23
  RUN --mount=type=secret,id=SECRET_KEY,mode=0444,required=true
24
  RUN --mount=type=secret,id=ALGORITHM,mode=0444,required=true
 
17
 
18
  COPY --chown=user . $HOME/app
19
 
20
+ ENV HF_HOME=/data/.huggingface
21
+
22
+ ENV DATABASE_URL=sqlite:///data/sql_app.db
23
+ ENV DEVICE=cuda
24
 
25
  RUN --mount=type=secret,id=SECRET_KEY,mode=0444,required=true
26
  RUN --mount=type=secret,id=ALGORITHM,mode=0444,required=true